def plan_shapes

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

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

80def plan_shapes(+target: Shape.Shape, +shape: Shape.Shape, +plan: Shape.BroadcastPlan,
81  planned: {Shape.broadcast_checked(target, shape) == Done{plan} : Result<&2, &2, Shape.Error, Shape.BroadcastPlan>}) ->
82  {(Shape.left_shape(plan), Shape.right_shape(plan)) == (target, shape) : Shape.Shape & Shape.Shape}:
83  %planned : {planned_shapes(target, shape, _) == (target, shape) : Shape.Shape & Shape.Shape}
84  shapes_checked(target, shape)

Uses (11)

Used by (2)