def Logical

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

import Base
import bendtorch-core@0.1.0.0/src/tensor/matmul.bend as Matrix
import bendtorch-core@0.1.0.0/src/tensor/dot_fold.bend as DotFold

200# The factor pairs k, k + 1, ... of cell (b1, b2, row, column): the left
201# operand's word at its term k of row `row`, the right operand's at its
202# term k of column `column`, each at its own batch.
203def Logical(+fuel: Nat, +dl: Nat, +dr: Nat, -A: Array<F32>, -B: Array<F32>, +lo: Matrix.Order, +ro: Matrix.Order, +inner: Nat,
204  +rows: Nat, +columns: Nat, +l1: Nat, +l2: Nat, +r1: Nat, +r2: Nat, +b1: Nat, +b2: Nat, +row: Nat, +column: Nat, +k: Nat,
205  values: List<&2, DotFold.Factors>) -> Data:
206  match fuel values:
207    case 0n Nil{}:
208      Unit
209    case 1n+tail DotFold.Factors{lv, rv} <> rest:
210      LogicalStep<tail, dl, dr, A, B, lo, ro, inner, rows, columns, l1, l2, r1, r2, b1, b2, row, column, k, lv, rv, rest>
211    case 0n DotFold.Factors{lv, rv} <> rest:
212      Empty
213    case 1n+tail Nil{}:
214      Empty