Method constructor

tfjs-backend-webgpu/src/diag_webgpu.ts:30–37

(size: number)

  size = true;

  constructor(size: number) {
    this.outputShape = [size, size];
    this.dispatchLayout = flatDispatchLayout(this.outputShape);
    this.dispatch = computeDispatch(
        this.dispatchLayout, this.outputShape, this.workgroupSize);

    this.shaderKey = 'diag';
  }

  getUserCode(): string {
    const userCode = `