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

Method constructor

tfjs-backend-webgpu/src/lin_space_webgpu.ts:31–38  ·  view source on GitHub ↗
(shape: number)

Source from the content-addressed store, hash-verified

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

Callers

nothing calls this directly

Calls 2

flatDispatchLayoutFunction · 0.90
computeDispatchFunction · 0.90

Tested by

no test coverage detected