Function reduce

tfjs-backend-webgl/src/kernel_utils/reduce.ts:46–76

(
    x: TensorInfo, dtype: DataType, reductionType: ReduceTypes,
    backend: MathBackendWebGL)

}

export function reduce(
    x: TensorInfo, dtype: DataType, reductionType: ReduceTypes,
    backend: MathBackendWebGL): TensorInfo {
  const reductionStages = getReductionStages(x.shape);

  let result = x;
  for (let i = 0; i < reductionStages.length; i++) {
    const {inSize, windowSize, outSize} = reductionStages[i];

    let program: ReduceProgram|MeanProgram;
    let previousResult: TensorInfo;
    if (reductionType === 'mean') {
      program = i === 0 ?
          new MeanProgram(
              {windowSize, inSize, batchSize: x.shape[0], outSize}, inSize) :
          new MeanProgram({windowSize, inSize, batchSize: x.shape[0], outSize});
    } else {
      program = new ReduceProgram(
          {windowSize, inSize, batchSize: x.shape[0], outSize}, reductionType);
    }

    previousResult = result;
    result = backend.runWebGLProgram(program, [result], dtype);

    if (previousResult.dataId !== x.dataId) {
      backend.disposeIntermediateTensorInfo(previousResult);
    }
  }

  return result;
}