Function flatDispatchLayout

tfjs-backend-webgpu/src/webgpu_util.ts:142–144

(shape: number[])

}

export function flatDispatchLayout(shape: number[]) {
  return {x: shape.map((d, i) => i)};
}

export function GPUBytesPerElement(dtype: DataType): number {
  if (dtype === 'float32' || dtype === 'int32' || dtype === 'bool' ||