def plan_unique

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

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

138def plan_unique(+input: Shape.Shape, +axes: List<&2, U32>, decision: Shape.BoolCase<unique(axes)>) ->
139  Result<&2, &2, Error, Plan<input, axes>>:
140  match decision:
141    case Shape.BFalse{equal}:
142      Fail{DuplicateAxis{}}
143    case Shape.BTrue{equal}:
144      plan_decided(input, axes, Shape.bool_case(bijective(input, axes)))

Uses (11)

Used by (6)