def Mat.gemm_t

bend-blas-lapack@0.0.0.1 · mat.bend:147

import Base
import bend-blas-lapack@0.0.0.1/blas.bend as B

147# C = alpha A B^T + beta C, B stored cols x k
148def Mat.gemm_t(+rows: Nat, +k: Nat, +cols: Nat, alpha: F32, a: Mat<rows, k>, b: Mat<cols, k>, beta: F32, c: Mat<rows, cols>) -> IO(Result<&1, &1, U32 & String, Mat<rows, cols>>):
149  MatA{x} = a
150  MatA{y} = b
151  MatA{z} = c
152  IO.bind(Result<&1, &1, U32 & String, Array<F32>>, Result<&1, &1, U32 & String, Mat<rows, cols>>, B.Blas.sgemm(0, 1, U32.from_nat(rows), U32.from_nat(cols), U32.from_nat(k), alpha, x, y, beta, z), r => Mat.wrap(rows, cols, r))