Interface WebGPUProgram

tfjs-backend-webgpu/src/webgpu_program.ts:27–59

}

export interface WebGPUProgram {
  // Whether to use atomic built-in functions.
  atomic?: boolean;
  // dispatch specifies geometry of thread groups - derived from dispatchLayout.
  dispatch: [number, number, number];
  // dispatchLayout enumerates how tensor dimensions are distributed among
  // dispatch x,y,z dimensions.
  dispatchLayout: {x: number[], y?: number[], z?: number[]};
  // By default, the output data component is 1.
  outputComponent?: number;
  outputShape: number[];
  pixelsOpType?: PixelsOpType;
  // The unique key to distinguish different shader source code.
  shaderKey: string;
  // Whether to use output size for bounds checking.
  size?: boolean;
  uniforms?: string;
  variableNames: string[];
  // Describe each variable's component and must have one-one mapping with
  // variableNames. If not set, all variables component will be same with output
  // component member.
  variableComponents?: number[];
  // workgroupSize.x * workgroupSize.y * workgroupSize.z = the number of threads
  // in a thread group. Individual dimensions determines thread layout within
  // the group.
  workgroupSize: [number, number, number];
  // Size of register cache in one dimension (assumes square cache).
  // Each thread writes to workPerThread * workPerThread locations in the output
  // buffer.
  workPerThread?: number;
  pipeline?: GPUComputePipeline|Promise<GPUComputePipeline>;
  getUserCode: () => string;
}

export const compileProgram =
    (device: GPUDevice, program: WebGPUProgram, inputsData: InputInfo[],