~/jendhub

def gemm_any

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

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

234# TA.gemm_any's arguments and result; one row runs the scalar loop at every depth, as in tensor-array 0.1.12.0
235def gemm_any(+ac: 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:
236  gemm_any2(ac, Nat.is_le(n, 1n), d, n, m, k, arow, sa, sb, bcol, a, b, c)

Uses (8)

Used by (7)