def planned_shapes

bendtorch-core@0.1.0.0 · src/autograd/reduce_to_fold_PROOF.bend:32

import bendtorch-core@0.1.0.0/src/autograd/reduce_to_fold_PROOF.bend as ReduceToFoldPROOF
import Base
import bendtorch-core@0.1.0.0/src/shape/core.bend as Shape

32def planned_shapes(left: Shape.Shape, right: Shape.Shape, result: Result<&2, &2, Shape.Error, Shape.BroadcastPlan>) -> Shape.Shape & Shape.Shape:
33  match result:
34    case Fail{error}:
35      (left, right)
36    case Done{+plan}:
37      (Shape.left_shape(plan), Shape.right_shape(plan))

Uses (8)

Used by (4)