def scatter_build

bendtorch-core@0.1.0.0 · src/tensor/gather.bend:349

import Base
import bendtorch-core@0.1.0.0/src/tensor/parallel.bend as Parallel

349def scatter_build(+depth: Nat, +splits: Nat, +plan: Plan, +count: Nat, +base: Nat, g: Array<F32>, index: Array<U32>) -> Array<F32>:
350  Parallel.build(~Plan, ~Plan, ~Array<F32>, ~Array<U32>, ~same, ~split, ~scatter_cell, plan, depth, splits, count, base, (g, index))