host_gemm.hpp File Reference#
host_gemm.hpp File Reference
#include "host_tensor.hpp"Go to the source code of this file.
Functions | |
| template<typename AType, typename BType, typename CType, typename AElementwiseOperation, typename BElementwiseOperation, typename CElementwiseOperation> | |
| void | host_gemm_mk_kn_mn (const Tensor< AType > &a_m_k, const Tensor< BType > &b_k_n, Tensor< CType > &c_m_n, const AElementwiseOperation &a_element_op, const BElementwiseOperation &b_element_op, const CElementwiseOperation &c_element_op) |
Function Documentation
◆ host_gemm_mk_kn_mn()
template<typename AType, typename BType, typename CType, typename AElementwiseOperation, typename BElementwiseOperation, typename CElementwiseOperation>
| void host_gemm_mk_kn_mn | ( | const Tensor< AType > & | a_m_k, |
| const Tensor< BType > & | b_k_n, | ||
| Tensor< CType > & | c_m_n, | ||
| const AElementwiseOperation & | a_element_op, | ||
| const BElementwiseOperation & | b_element_op, | ||
| const CElementwiseOperation & | c_element_op ) |