#include "tensor/tensor.h"
#include "cblas.h"
Go to the source code of this file.
|
#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) |
|
|
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) |
|
- Author
- Daniel Nichols
- Version
- 0.1
- Date
- 2019-06-06
- Copyright
- Copyright (c) 2019