def plan_decided

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

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

130def plan_decided(+input: Shape.Shape, +axes: List<&2, U32>, decision: Shape.BoolCase<bijective(input, axes)>) ->
131  Result<&2, &2, Error, Plan<input, axes>>:
132  match decision:
133    case Shape.BFalse{equal}:
134      Fail{MappingError{}}
135    case Shape.BTrue{equal}:
136      plan_selected(input, axes, equal, reorder(Shape.dimensions(input), axes), {==})

Uses (11)

Used by (2)