def built_array

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

import Base
import bendtorch-core@0.1.0.0/src/tensor/slice.bend as Slice

362def built_array(kind: Pass, +depth: Nat, +plan: Plan, +count: Nat, a: Array<F32>, index: Array<U32>) -> Array<F32>:
363  match kind:
364    case Forward{}:
365      build(depth, Slice.splits(depth), plan, count, 0n, a, index)
366    case Backward{}:
367      scatter_build(depth, Slice.splits(depth), plan, count, 0n, a, index)