Function sparseSegmentSum

tfjs-backend-wasm/src/kernels/SparseSegmentSum.ts:24–29

(args: {
  backend: BackendWasm,
  inputs: SparseSegmentSumInputs,
})

import {setup, sparseSegmentReduction} from './SparseSegmentReduction';

function sparseSegmentSum(args: {
  backend: BackendWasm,
  inputs: SparseSegmentSumInputs,
}): TensorInfo {
  return sparseSegmentReduction(args, false);
}

export const sparseSegmentSumConfig: KernelConfig = {
  kernelName: SparseSegmentSum,