def allocated_dimensions

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

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.bend as Elementwise
import bendtorch-core@0.1.0.0/src/tensor/elementwise_totality.bend as Totality

494def allocated_dimensions(+operation: Elementwise.Operation, +plan: Shape.BroadcastPlan, left: Core.Dynamic, right: Core.Dynamic,
495  result: Result<&2, &1, Core.Error, Core.Tensor<Shape.output_shape(plan)>>) ->
496  Totality.DimensionsModel<Shape.dimensions(Shape.output_shape(plan)), Elementwise.allocated(operation, plan, left, right, result)>:
497  match result:
498    case Fail{error}:
499      Totality.DimensionsModel{Unit{}}
500    case Done{output}:
501      loop_dimensions(Shape.elements(Shape.output_shape(plan)), operation, plan, 0n, Elementwise.State{left, right, output})