Method constructor

tfjs-backend-webgpu/src/gather_webgpu.ts:31–39

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

  size = true;

  constructor(aShape: number[], outputShape: number[]) {
    this.outputShape = aShape.slice();
    this.aShape = aShape;
    this.outputShape = outputShape;
    this.dispatchLayout = flatDispatchLayout(this.outputShape);
    this.dispatch = computeDispatch(
        this.dispatchLayout, this.outputShape, this.workgroupSize);
    this.shaderKey = `gather`;
  }

  getUserCode(): string {
    const sourceCoords = getSourceCoords(this.aShape);