def mm_nt

bend-ml-tensor-array@0.1.19.0 · main.bend:1222

import bend-ml-tensor-array@0.1.19.0/main.bend as BendMlTensorArray
import Base

1222def mm_nt(+n: Nat, +k: Nat, +m: Nat, d: Nat, a: Array<F32>, b: Array<F32>) -> MMulNT<n, k, m>:
1223  mmul_nt_fin(n, k, m, gemm_any(False{}, d, n, m, k, U32.from_nat(k), 1, 1, U32.from_nat(k), a, b, Array.new(F32, cap_depth(Nat.mul(n, m)), 0.0)))

Uses (11)

Used by (1)