def Mat.gemm_tn

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

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

157# C = alpha A^T B + beta C, A stored k x rows
158def Mat.gemm_tn(+rows: Nat, +k: Nat, +cols: Nat, alpha: F32, a: Mat<k, rows>, b: Mat<k, cols>, beta: F32, c: Mat<rows, cols>) -> IO(Result<&1, &1, U32 & String, Mat<rows, cols>>):
159  MatA{x} = a
160  MatA{y} = b
161  MatA{z} = c
162  IO.bind(Result<&1, &1, U32 & String, Array<F32>>, Result<&1, &1, U32 & String, Mat<rows, cols>>, B.Blas.sgemm(1, 0, U32.from_nat(rows), U32.from_nat(cols), U32.from_nat(k), alpha, x, y, beta, z), r => Mat.wrap(rows, cols, r))