def retained_fits

bendtorch-core@0.1.0.0 · src/tensor/elementwise_PROOF.bend:773

import Base
import bendtorch-core@0.1.0.0/src/shape/core.bend as Shape
import bendtorch-core@0.1.0.0/src/tensor/core.bend as Core
import bendtorch-core@0.1.0.0/src/tensor/elementwise_totality.bend as Totality

773def retained_fits(+plan: Shape.BroadcastPlan, -before_left: Core.Dynamic, -before_right: Core.Dynamic,
774  -after_left: Core.Dynamic, -after_right: Core.Dynamic,
775  -left_same: Totality.SameSource(before_left, after_left), -right_same: Totality.SameSource(before_right, after_right),
776  -fits: FittingSources<plan, before_left, before_right>) -> FittingSources<plan, after_left, after_right>:
777  FittingSources{
778    Equal.trans(List<&2, U32>, Totality.dimensions(after_left), Totality.dimensions(before_left), Shape.dimensions(Shape.left_shape(plan)),
779      Equal.cong(Totality.Inspection, List<&2, U32>, inspection_dimensions,
780        Totality.inspect(Totality.Dimensions{}, after_left), Totality.inspect(Totality.Dimensions{}, before_left), left_same(Totality.Dimensions{})),
781      {\{FittingSources: l => r => l} : FittingSources<plan, before_left, before_right> -> {Totality.dimensions(before_left) == Shape.dimensions(Shape.left_shape(plan)) : List<&2, U32>}}(fits)),
782    Equal.trans(List<&2, U32>, Totality.dimensions(after_right), Totality.dimensions(before_right), Shape.dimensions(Shape.right_shape(plan)),
783      Equal.cong(Totality.Inspection, List<&2, U32>, inspection_dimensions,
784        Totality.inspect(Totality.Dimensions{}, after_right), Totality.inspect(Totality.Dimensions{}, before_right), right_same(Totality.Dimensions{})),
785      {\{FittingSources: l => r => r} : FittingSources<plan, before_left, before_right> -> {Totality.dimensions(before_right) == Shape.dimensions(Shape.right_shape(plan)) : List<&2, U32>}}(fits))}