def plan_right

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

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

92def plan_right(+target: Shape.Shape, +shape: Shape.Shape, +plan: Shape.BroadcastPlan,
93  planned: {Shape.broadcast_checked(target, shape) == Done{plan} : Result<&2, &2, Shape.Error, Shape.BroadcastPlan>}) ->
94  {Shape.right_shape(plan) == shape : Shape.Shape}:
95  Equal.cong(Shape.Shape & Shape.Shape, Shape.Shape, second, (Shape.left_shape(plan), Shape.right_shape(plan)), (target, shape),
96    plan_shapes(target, shape, plan, planned))

Uses (12)

Used by (2)