Method getUserCode

tfjs-backend-webgpu/src/conv2d_mm_webgpu.ts:250–269

()

  }

  getUserCode(): string {
    const matMulSource = this.isVec4 ?
        makeMatMulPackedVec4Source(
            this.elementsPerThread, this.workgroupSize, !this.isChannelsLast,
            this.tileInner) :
        makeMatMulPackedSource(
            this.elementsPerThread, this.workgroupSize, !this.isChannelsLast,
            this.tileInner, false, null, this.sequentialAccessByThreads);
    const elementsSize =
        this.isVec4 ? [this.innerElementSize, 4, 4] : [1, 1, 1];
    const userCode = `
    ${
        conv2dCommonSnippet(
            this.isChannelsLast, this.fitAOuter, this.fitBOuter, this.fitInner,
            this.addBias, this.activation, this.hasPreluActivationWeights,
            elementsSize[0], elementsSize[1], elementsSize[2])}
    ${matMulSource}
  `;
    return userCode;
  }
}