Method constructor

tfjs-backend-webgpu/src/reverse_webgpu.ts:31–38

(xShape: [number, number, number, number])

  size = true;

  constructor(xShape: [number, number, number, number]) {
    this.outputShape = xShape;
    this.dispatchLayout = flatDispatchLayout(this.outputShape);
    this.dispatch = computeDispatch(
        this.dispatchLayout, this.outputShape, this.workgroupSize);
    this.uniforms = ` axis : vec4<i32>,`;
    this.shaderKey = 'reverse';
  }

  getUserCode(): string {
    const reverseCoordsSnippet = `