#include <TensorContraction.h>
Public Member Functions | |
template<typename Index , typename Scalar > | |
EIGEN_ALWAYS_INLINE void | operator() (const internal::blas_data_mapper< Scalar, Index, ColMajor > &output_mapper, const TensorContractionParams ¶ms, Index i, Index j, Index num_rows, Index num_cols) const |
Definition at line 294 of file TensorContraction.h.
|
inline |
Tensor contraction evaluator calls this kernel after finishing each block of output matrix. Output blocks belong to the 2-dimensional output tensor.
TensorContractionParams contains contraction dimensions information required to map output 2-d space into the expected output tensor space (potentially higher dimensional).
[in] | output_mapper | Access to output tensor memory |
[in] | params | Tensor contraction parameters |
[in] | i | Index of a first row available through output_mapper |
[in] | j | Index of a first column available through output_mapper |
[in] | num_rows | Number of available rows |
[in] | num_cols | Number of available columns |
Definition at line 311 of file TensorContraction.h.