Method constructor

tfjs-backend-webgl/src/diag_gpu.ts:25–34

(size: number)

  userCode: string;

  constructor(size: number) {
    this.outputShape = [size, size];
    this.userCode = `
      void main() {
          ivec2 coords = getOutputCoords();
          float val = coords[0] == coords[1] ? getX(coords[0]) : 0.0;
          setOutput(val);
      }
    `;
  }
}