Method constructor

tfjs-backend-webgpu/src/tile_webgpu.ts:31–42

(aShape: number[], reps: number[])

  rank: number;

  constructor(aShape: number[], reps: number[]) {
    const outputShape: number[] = new Array(aShape.length);
    for (let i = 0; i < outputShape.length; i++) {
      outputShape[i] = aShape[i] * reps[i];
    }
    this.outputShape = outputShape;
    this.dispatchLayout = flatDispatchLayout(this.outputShape);
    this.dispatch = computeDispatch(
        this.dispatchLayout, this.outputShape, this.workgroupSize);
    this.rank = this.outputShape.length;
    this.shaderKey = 'tile';
  }

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