~/jendhub

def output_dimensions

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

import bendtorch-core@0.1.0.0/src/tensor/elementwise_PROOF.bend as ElementwisePROOF
import Base
import bendtorch-core@0.1.0.0/src/shape/core.bend as Shape
import bendtorch-core@0.1.0.0/src/storage/main.bend as Storage
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

534def output_dimensions(+operation: Elementwise.Operation, +plan: Shape.BroadcastPlan, +depth: Nat, +length: Nat,
535  -valid: {Storage.bounds(depth, length) == True{} : Bool},
536  -exact: {length == Shape.logical_numel(Shape.dimensions(Shape.output_shape(plan))) : Nat},
537  taken_left: Core.Dynamic & Array<F32>, taken_right: Core.Dynamic & Array<F32>) ->
538  Totality.DimensionsModel<Shape.dimensions(Shape.output_shape(plan)), Elementwise.parallel_output(operation, plan, depth, length, valid, exact, taken_left, taken_right)>:
539  (kept_left, la) = taken_left
540  (kept_right, ra) = taken_right
541  Totality.DimensionsModel{{==}}

Uses (15)

Used by (1)