Method constructor

tfjs-backend-webgpu/src/onehot_webgpu.ts:31–37

(numIndices: number, depth: number)

  size = true;

  constructor(numIndices: number, depth: number) {
    this.outputShape = [numIndices, depth];
    this.dispatchLayout = flatDispatchLayout(this.outputShape);
    this.dispatch = computeDispatch(
        this.dispatchLayout, this.outputShape, this.workgroupSize);
    this.shaderKey = 'onehot';
  }

  getUserCode(): string {
    const userCode = `