~/jendhub

def mm_tn

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

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

1239def mm_tn(+n: Nat, +k: Nat, +m: Nat, d: Nat, a: Array<F32>, b: Array<F32>) -> MMulTN<n, k, m>:
1240  mmul_tn_fin(n, k, m, gemm_any(False{}, d, n, m, k, 1, U32.from_nat(n), U32.from_nat(m), 1, a, b, Array.new(F32, cap_depth(Nat.mul(n, m)), 0.0)))

Uses (11)

Used by (1)