def planned_fiber

bendtorch-core@0.1.0.0 · src/autograd/reduce_LAWS.bend:60

import Base
import bendtorch-core@0.1.0.0/src/shape/core.bend as Shape

60def planned_fiber(+values: List<&2, F32>, +target_index: U32, +count: Nat,
61  plan: Result<&2, &2, Shape.Error, Shape.BroadcastPlan>) -> Maybe<&2, F32>:
62  match plan:
63    case Fail{error}:
64      None{}
65    case Done{+plan}:
66      fiber(plan, values, target_index, count)