module testing.einsum.blas_lapack#

Short summary#

module mlprodict.testing.einsum.blas_lapack

Direct calls to libraries BLAS and LAPACK.

source on GitHub

Functions#

function

truncated documentation

gemm_dot

Implements dot product with gemm when possible.

pygemm

Pure python implementatin of GEMM.

Documentation#

Direct calls to libraries BLAS and LAPACK.

source on GitHub

mlprodict.testing.einsum.blas_lapack.gemm_dot(A, B, transA=False, transB=False)#

Implements dot product with gemm when possible.

Parameters:

  • A – first matrix

  • B – second matrix

  • transA – is first matrix transposed?

  • transB – is second matrix transposed?

source on GitHub

mlprodict.testing.einsum.blas_lapack.pygemm(transA, transB, M, N, K, alpha, A, lda, B, ldb, beta, C, ldc)#

Pure python implementatin of GEMM.

source on GitHub