Function isFlatDispatch

tfjs-backend-webgpu/src/webgpu_program.ts:778–780

(program: WebGPUProgram)

}

function isFlatDispatch(program: WebGPUProgram): boolean {
  return program.dispatch[1] === 1 && program.dispatch[2] === 1;
}

export function dataTypeToGPUType(type: DataType, component = 1) {
  if (type === 'float32') {