Method constructor

tfjs-backend-webgl/src/slice_gpu.ts:28–51

(destSize: number[])

  customUniforms: Array<{name: string; arrayIndex: number; type: UniformType;}>;

  constructor(destSize: number[]) {
    this.outputShape = destSize;
    this.rank = destSize.length;

    const dtype = getCoordsDataType(this.rank);
    this.customUniforms = [{name: 'start', arrayIndex: this.rank, type: 'int'}];
    const sourceCoords = getCoords(this.rank);

    let body: string;
    const coordSum = destSize.map((_, i) => {
      return `sourceLoc.${coords[i]} = start[${i}] + coords.${coords[i]};`;
    });
    body = `
        ${dtype} sourceLoc;
        ${dtype} coords = getOutputCoords();
        ${coordSum.join('\n')}
      `;
    this.userCode = `
      void main() {
        ${body}
        setOutput(getSource(${sourceCoords}));
      }
    `;
  }
}

const coords = ['x', 'y', 'z', 'w', 'u', 'v'];