def gemm_ac

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

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

import Base

107def gemm_ac(+ac: Bool, +n: Nat, +m: Nat, +k: Nat, +arow: U32, +sa: U32, +sb: U32, +bcol: U32, a: Array<F32>, b: Array<F32>, c: Array<F32>) -> G:
108  grows(ac, n, Nat.sub(m, 1n), k, 0, arow, sa, sb, bcol, U32.from_nat(m), G{a, b, c})

Uses (9)

Used by (4)