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

Method constructor

tfjs-backend-webgpu/src/cum_webgpu.ts:40–51  ·  view source on GitHub ↗
(
      op: CumOpType, shape: number[], exclusive: boolean, reverse: boolean)

Source from the content-addressed store, hash-verified

38 op: CumOpType;
39
40 constructor(
41 op: CumOpType, shape: number[], exclusive: boolean, reverse: boolean) {
42 this.workgroupSize = [128, 1, 1];
43 this.outputShape = shape;
44 this.dispatchLayout = flatDispatchLayout(this.outputShape);
45 this.dispatch = computeDispatch(
46 this.dispatchLayout, this.outputShape, this.workgroupSize);
47 this.exclusive = exclusive;
48 this.reverse = reverse;
49 this.op = op;
50 this.shaderKey = `cum_${this.op}_${this.exclusive}_${this.reverse}`;
51 }
52
53 getUserCode(): string {
54 const rank = this.outputShape.length;

Callers

nothing calls this directly

Calls 2

flatDispatchLayoutFunction · 0.90
computeDispatchFunction · 0.90

Tested by

no test coverage detected