def shapes_construct

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

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

39def shapes_construct(+left: Shape.Shape, +right: Shape.Shape, -left_padding: Nat, -right_padding: Nat,
40  -minimal: Shape.PadMinimal<left_padding, right_padding>, -output: List<&2, U32>,
41  -aligned: Shape.Aligned(Shape.pad(left_padding, Shape.extents(left)), Shape.pad(right_padding, Shape.extents(right)), Shape.naturals(output)),
42  +result: Shape.ConstructResult<output>) ->
43  {planned_shapes(left, right, Shape.broadcast_checked.construct(left, right, left_padding, right_padding, minimal, output, aligned, result)) ==
44    (left, right) : Shape.Shape & Shape.Shape}:
45  match result:
46    case Shape.SError{error}:
47      {==}
48    case Shape.SBuilt{shape, preserved}:
49      {==}

Uses (14)

Used by (1)