def right_extents

bendtorch-core@0.1.0.0 · src/tensor/matmul_PROOF.bend:1372

import bendtorch-core@0.1.0.0/src/tensor/matmul_PROOF.bend as MatmulPROOF
import Base
import bendtorch-core@0.1.0.0/src/shape/core.bend as Shape

1372def right_extents(+right: Shape.Shape, +inner: U32, +right_inner: U32, +columns: U32,
1373  +re: {Shape.dimensions(right) == [right_inner, columns] : List<&2, U32>},
1374  +compatible: {U32.to_nat(inner) == U32.to_nat(right_inner) : Nat}) ->
1375  {Shape.extents(right) == [U32.to_nat(inner), U32.to_nat(columns)] : List<&2, Nat>}:
1376  Equal.trans(List<&2, Nat>, Shape.extents(right), [U32.to_nat(right_inner), U32.to_nat(columns)], [U32.to_nat(inner), U32.to_nat(columns)],
1377    Equal.cong(List<&2, U32>, List<&2, Nat>, dims => Shape.naturals(dims), Shape.dimensions(right), [right_inner, columns], re),
1378    Equal.cong(Nat, List<&2, Nat>, n => [n, U32.to_nat(columns)], U32.to_nat(right_inner), U32.to_nat(inner),
1379      Equal.sym(Nat, U32.to_nat(inner), U32.to_nat(right_inner), compatible)))

Uses (11)

Used by (1)