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

Method constructor

tfjs-backend-webgpu/src/transform_webgpu.ts:31–37  ·  view source on GitHub ↗
(outShape: [number, number, number, number])

Source from the content-addressed store, hash-verified

29 size = true;
30
31 constructor(outShape: [number, number, number, number]) {
32 this.outputShape = outShape;
33 this.dispatchLayout = flatDispatchLayout(this.outputShape);
34 this.dispatch = computeDispatch(
35 this.dispatchLayout, this.outputShape, this.workgroupSize);
36 this.shaderKey = 'transform';
37 }
38
39 getUserCode(): string {
40 const userCode = `

Callers

nothing calls this directly

Calls 2

flatDispatchLayoutFunction · 0.90
computeDispatchFunction · 0.90

Tested by

no test coverage detected