Function isFlatDispatchLayout

tfjs-backend-webgpu/src/webgpu_program.ts:843–853

(program: WebGPUProgram)

  return uniformShader;
}
function isFlatDispatchLayout(program: WebGPUProgram): boolean {
  if (program.dispatchLayout.hasOwnProperty('y') &&
      program.dispatchLayout.y.length !== 0) {
    return false;
  }
  if (program.dispatchLayout.hasOwnProperty('z') &&
      program.dispatchLayout.z.length !== 0) {
    return false;
  }
  return true;
}