def parallel_planned

bendtorch-core@0.1.0.0 · src/tensor/elementwise.bend:492

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

492def parallel_planned(operation: Operation, left: Core.Dynamic, right: Core.Dynamic,
493  result: Result<&2, &2, Shape.Error, Shape.BroadcastPlan>) ->
494  (Core.Dynamic & Core.Dynamic) & Result<&2, &1, Core.Error, Core.Dynamic>:
495  match result:
496    case Fail{error}:
497      ((left, right), Fail{Core.ShapeError{error}})
498    case Done{+plan}:
499      parallel_allocated(operation, plan, left, right, Core.new(Shape.output_shape(plan), 0.0))