Method getUserCode

tfjs-backend-webgpu/src/tile_webgpu.ts:44–56

()

  }

  getUserCode(): string {
    const sourceCoords = getSourceCoords(this.rank, 'uniforms.');

    const userCode = `
      ${main('index')} {
        if (index < uniforms.size) {
          let resRC = getCoordsFromIndex(index);
          setOutputAtIndex(index, getA(${sourceCoords}));
        }
      }
    `;
    return userCode;
  }
}

function getSourceCoords(rank: number, uniformPrefix = ''): string {