Method constructor

tfjs-backend-webgl/src/onehot_gpu.ts:28–40

(
      numIndices: number, depth: number, onValue: number, offValue: number)

  seedLoc: WebGLUniformLocation;

  constructor(
      numIndices: number, depth: number, onValue: number, offValue: number) {
    this.outputShape = [numIndices, depth];

    this.userCode = `
      void main() {
        ivec2 coords = getOutputCoords();
        int index = round(getIndices(coords.x));
        setOutput(mix(float(${offValue}), float(${onValue}),
                      float(index == coords.y)));
      }
    `;
  }
}