MagmaDNN  1.0
c++NeuralNetworkFramework
matmul.h File Reference
#include "tensor/tensor.h"
#include "cblas.h"
Include dependency graph for matmul.h:
This graph shows which files directly or indirectly include this file:

Go to the source code of this file.

Macros

#define MAGMA_SGEMM_ROWMAJOR(A, B, C, m, n, k, alpha, beta, transf_A, transf_B, lda, ldb, ldc)   magma_sgemm(transf_B, transf_A, n, m, k, alpha, B, ldb, A, lda, beta, C, ldc)
 
#define MAGMA_DGEMM_ROWMAJOR(A, B, C, m, n, k, alpha, beta, transf_A, transf_B, lda, ldb, ldc)   magma_dgemm(transf_B, transf_A, n, m, k, alpha, B, ldb, A, lda, beta, C, ldc)
 

Functions

template<typename T >
void magmadnn::math::matmul (T alpha, bool trans_A, Tensor< T > *A, bool trans_B, Tensor< T > *B, T beta, Tensor< T > *C)
 

Detailed Description

Author
Daniel Nichols
Version
0.1
Date
2019-06-06