def plan_checked

bendtorch-core@0.1.0.0 · src/autograd/sum_dims.bend:96

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

96def plan_checked(+axes: List<&2, U32>, +dimensions: List<&2, U32>, checked: Maybe<&2, Layout.Error>) -> Result<&2, &2, Error, Plan>:
97  match checked:
98    case Some{error}:
99      Fail{TransformError{Transforms.LayoutError{error}}}
100    case None{}:
101      plan_built(kept_dimensions(axes, dimensions), dropped_dimensions(axes, dimensions),
102        Shape.construct(kept_dimensions(axes, dimensions)))