def mp_s3

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

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

281def mp_s3(+ac: Bool, +par: Nat, +n: Nat, +m: Nat, +w: Nat, +w32: U32, +m32: U32, a: Array<F32>, as: Array<F32>, c: Array<F32>, sb: TA.SL) -> TA.G:
282  match sb:
283    case TA.SL{b, bs}: TA.mp_s4(a, b, gemm_any(ac, TA.split_depth(par, n, m, w), n, m, w, w32, 1, m32, 1, as, bs, c))
284
285@unsafe

Uses (10)

Used by (1)

bend-ml-tensor-fork@0.1.6.0/main:mp_s2