def gemm

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

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

import Base

273def gemm(+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>) -> ABC:
274  run_e(e_of(ac, n, m, k, arow, sa, sb, bcol), U32.from_nat(tiles(n, m)), a, b, c)

Uses (10)

Used by (3)