Function getWorkgroupSizeString

tfjs-backend-webgpu/src/webgpu_program.ts:191–196

(program: WebGPUProgram)

}

export function getWorkgroupSizeString(program: WebGPUProgram): string {
  return `
  @compute @workgroup_size(${program.workgroupSize[0]}, ${
      program.workgroupSize[1]}, ${program.workgroupSize[2]})
`;
}

function makeShader(
    inputInfo: InputInfo[], outputData: {dtype: DataType, shape: number[]},