def matmul_tn_acc

bend-ml-tensor-gpu@0.1.5.1 · main.bend:325

import bend-ml-tensor-gpu@0.1.5.1/main.bend as BendMlTensorGpu

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

325def matmul_tn_acc(+n: Nat, +k: Nat, +m: Nat, par: Nat, a: TA.Mat<k, n>, b: TA.Mat<k, m>, c: TA.Mat<n, m>) -> TA.MMulTN<n, k, m>:
326  match a b c:
327    case TA.Mat{x} TA.Mat{y} TA.Mat{z}:
328      mm_tn_fin(n, k, m, gemm(True{}, n, m, k, 1, U32.from_nat(n), U32.from_nat(m), 1, x, y, z))

Uses (7)

Used by (3)