Method constructor

tfjs-backend-webgpu/src/pool_webgpu.ts:39–58

(
      convInfo: backend_util.Conv2DInfo, poolType: 'max'|'avg',
      computePositions = false, flattenPositions = false,
      includeBatchIndex = false)

  includeBatchIndex: boolean;

  constructor(
      convInfo: backend_util.Conv2DInfo, poolType: 'max'|'avg',
      computePositions = false, flattenPositions = false,
      includeBatchIndex = false) {
    if (poolType === 'avg' && computePositions) {
      throw new Error('Cannot compute positions for average pool.');
    }

    this.outputShape = convInfo.outShape;
    this.dispatchLayout = flatDispatchLayout(this.outputShape);
    this.dispatch = computeDispatch(
        this.dispatchLayout, this.outputShape, this.workgroupSize);

    this.poolType = poolType;
    this.computePositions = computePositions;
    this.flattenPositions = flattenPositions;
    this.includeBatchIndex = includeBatchIndex;
    this.shaderKey = `pool2D_${poolType}_${computePositions}_${
        flattenPositions}_${includeBatchIndex}`;
  }

  getUserCode(): string {
    let updateSnippet: string;