#include "cblas.h"
#include "tensor/tensor.h"
Go to the source code of this file.
- Author
- Daniel Nichols
- Version
- 1.0
- Date
- 2019-02-22
- Copyright
- Copyright (c) 2019
◆ 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
-
- Parameters
-
- Returns
- true
-
false
◆ gemm_full()
template<typename T >
void magmadnn::internal::gemm_full |
( |
T |
alpha, |
|
|
Tensor< T > * |
A, |
|
|
Tensor< T > * |
B, |
|
|
T |
beta, |
|
|
Tensor< T > * |
C |
|
) |
| |
Computes the matrix product C = alpha*(AB) + beta*C
- Template Parameters
-
- Parameters
-