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

Method constructor

tfjs-backend-webgpu/src/fft_webgpu.ts:32–40  ·  view source on GitHub ↗
(component: 'real'|'imag', shape: [number, number])

Source from the content-addressed store, hash-verified

30 component: string;
31
32 constructor(component: 'real'|'imag', shape: [number, number]) {
33 this.outputShape = shape;
34 this.dispatchLayout = flatDispatchLayout(this.outputShape);
35 this.dispatch = computeDispatch(
36 this.dispatchLayout, this.outputShape, this.workgroupSize);
37
38 this.component = component;
39 this.shaderKey = `fft_${component}`;
40 }
41
42 getUserCode(): string {
43 const opString = this.component === 'real' ?

Callers

nothing calls this directly

Calls 2

flatDispatchLayoutFunction · 0.90
computeDispatchFunction · 0.90

Tested by

no test coverage detected