MCPcopy Create free account
hub / github.com/tensorflow/tfjs / constructor

Method constructor

tfjs-backend-webgpu/src/gather_nd_webgpu.ts:31–39  ·  view source on GitHub ↗
(sliceDim: number, shape: number[])

Source from the content-addressed store, hash-verified

29 size = true;
30 sliceDim: number;
31 constructor(sliceDim: number, shape: number[]) {
32 this.outputShape = shape;
33 this.dispatchLayout = flatDispatchLayout(this.outputShape);
34 this.dispatch = computeDispatch(
35 this.dispatchLayout, this.outputShape, this.workgroupSize);
36 this.shaderKey = `gathernd_${sliceDim}`;
37 this.sliceDim = sliceDim;
38 this.uniforms = `sliceDim : i32, strides : ${getCoordsDataType(sliceDim)},`;
39 }
40
41 getUserCode(): string {
42 let strideString;

Callers

nothing calls this directly

Calls 3

flatDispatchLayoutFunction · 0.90
computeDispatchFunction · 0.90
getCoordsDataTypeFunction · 0.90

Tested by

no test coverage detected