MagmaDNN  1.0
c++NeuralNetworkFramework
gemm_internal.cpp File Reference
Include dependency graph for gemm_internal.cpp:

Functions

template<typename T >
bool magmadnn::internal::gemm_check (Tensor< T > *A, Tensor< T > *B, Tensor< T > *C, unsigned int &M, unsigned int &N, unsigned int &K)
 
template<>
void magmadnn::internal::gemm_full (int alpha, Tensor< int > *A, Tensor< int > *B, int beta, Tensor< int > *C)
 
template<>
void magmadnn::internal::gemm_full (float alpha, Tensor< float > *A, Tensor< float > *B, float beta, Tensor< float > *C)
 
template<>
void magmadnn::internal::gemm_full (double alpha, Tensor< double > *A, Tensor< double > *B, double beta, Tensor< double > *C)
 

Detailed Description

Author
Daniel Nichols
Version
1.0
Date
2019-02-22

Function Documentation

◆ gemm_check()

template<typename T >
bool magmadnn::internal::gemm_check ( Tensor< T > *  A,
Tensor< T > *  B,
Tensor< T > *  C,
unsigned int &  M,
unsigned int &  N,
unsigned int &  K 
)

Returns true if A, B, C are valid parameters for gemm_full. It also sets M, N, K to A.get_shape(0), B.get_shape(1), and A.get_shape(1), respectively.

Template Parameters
T
Parameters
A
B
C
M
N
K
Returns
true
false
Here is the call graph for this function:
Here is the caller graph for this function: