def layout_of

bendtorch-core@0.1.0.0 · src/tensor/batched_matmul_LAWS.bend:73

import Base
import bendtorch-core@0.1.0.0/src/tensor/matmul.bend as Matrix
import bendtorch-core@0.1.0.0/src/tensor/batched_matmul.bend as Batched

73# The cells of a build: the layout's fields, named so a law can take them.
74def layout_of(+lo: Matrix.Order, +ro: Matrix.Order, +inner: Nat, +rows: Nat, +columns: Nat,
75  +l1: Nat, +l2: Nat, +r1: Nat, +r2: Nat, +o2: Nat) -> Batched.Layout:
76  Batched.Layout{lo, ro, inner, rows, columns, l1, l2, r1, r2, o2}