def axis_choice.equal
bendtorch-core@0.1.0.0 · src/shape/core.bend:745
import bendtorch-core@0.1.0.0/src/shape/core.bend as Core
import Base
745def axis_choice.equal(+left: U32, +right: U32, same: BoolCase<Nat.is_eq(U32.to_nat(left), U32.to_nat(right))>) -> AxisChoice<left, right>:
746 match same:
747 case BFalse{equal}:
748 axis_choice.left_one(left, right, bool_case(Nat.is_eq(1n, U32.to_nat(left))))
749 case BTrue{equal}:
750 Picks{left, AxisEqual{{==}, equal_from(U32.to_nat(left), U32.to_nat(right), equal)}}