def plan_bounded

bendtorch-core@0.1.0.0 · src/tensor/layout.bend:146

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

146def plan_bounded(+input: Shape.Shape, +axes: List<&2, U32>, decision: Shape.BoolCase<bounded(axes, Shape.rank(input))>) ->
147  Result<&2, &2, Error, Plan<input, axes>>:
148  match decision:
149    case Shape.BFalse{equal}:
150      Fail{AxisOutOfBounds{}}
151    case Shape.BTrue{equal}:
152      plan_unique(input, axes, Shape.bool_case(unique(axes)))

Uses (12)

Used by (9)