def gemm_any2

bend-ml-tensor-fork@0.1.6.0 · main.bend:227

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

227def gemm_any2(+ac: Bool, one: Bool, d: Nat, +n: Nat, +m: Nat, +k: Nat, +arow: U32, +sa: U32, +sb: U32, +bcol: U32, a: Array<F32>, b: Array<F32>, c: Array<F32>) -> TA.G:
228  match one:
229    case True{}:
230      TA.gemm_ac(ac, n, m, k, arow, sa, sb, bcol, a, b, c)
231    case False{}:
232      gemm_rows(ac, d, n, m, k, arow, sa, sb, bcol, a, b, c)