Method constructor

tfjs-backend-webgpu/src/softmax_webgpu.ts:29–39

(outputShape: number[])

  workgroupSize: [number, number, number];

  constructor(outputShape: number[]) {
    this.outputShape = outputShape;  // [rows, cols]
    this.dispatchLayout = flatDispatchLayout(this.outputShape);
    this.dispatch = [this.outputShape[0], 1, 1];
    if (this.outputShape[1] >= 4096) {
      this.workgroupSize = [256, 1, 1];
    } else {
      this.workgroupSize = [64, 1, 1];
    }
    this.shaderKey = 'softmax';
  }

  getUserCode(): string {
    const userCode = `