#include <TensorContractionBlocking.h>
Public Types | |
typedef LhsMapper::Scalar | LhsScalar |
typedef RhsMapper::Scalar | RhsScalar |
Public Member Functions | |
EIGEN_DEVICE_FUNC EIGEN_ALWAYS_INLINE Index | kc () const |
EIGEN_DEVICE_FUNC EIGEN_ALWAYS_INLINE Index | mc () const |
EIGEN_DEVICE_FUNC EIGEN_ALWAYS_INLINE Index | nc () const |
EIGEN_DEVICE_FUNC | TensorContractionBlocking (Index k, Index m, Index n, Index num_threads=1) |
Private Attributes | |
Index | kc_ |
Index | mc_ |
Index | nc_ |
Definition at line 25 of file TensorContractionBlocking.h.
typedef LhsMapper::Scalar Eigen::internal::TensorContractionBlocking< LhsMapper, RhsMapper, Index, ShardingType >::LhsScalar |
Definition at line 28 of file TensorContractionBlocking.h.
typedef RhsMapper::Scalar Eigen::internal::TensorContractionBlocking< LhsMapper, RhsMapper, Index, ShardingType >::RhsScalar |
Definition at line 29 of file TensorContractionBlocking.h.
|
inline |
Definition at line 31 of file TensorContractionBlocking.h.
|
inline |
Definition at line 42 of file TensorContractionBlocking.h.
|
inline |
Definition at line 43 of file TensorContractionBlocking.h.
|
inline |
Definition at line 44 of file TensorContractionBlocking.h.
|
private |
Definition at line 47 of file TensorContractionBlocking.h.
|
private |
Definition at line 48 of file TensorContractionBlocking.h.
|
private |
Definition at line 49 of file TensorContractionBlocking.h.