#include <TensorContractionBlocking.h>
Public Member Functions | |
EIGEN_DEVICE_FUNC EIGEN_ALWAYS_INLINE StorageIndex | kc () const |
EIGEN_DEVICE_FUNC EIGEN_ALWAYS_INLINE StorageIndex | mc () const |
EIGEN_DEVICE_FUNC EIGEN_ALWAYS_INLINE StorageIndex | nc () const |
EIGEN_DEVICE_FUNC | TensorContractionBlocking (StorageIndex k, StorageIndex m, StorageIndex n, StorageIndex num_threads=1) |
Private Attributes | |
StorageIndex | kc_ |
StorageIndex | mc_ |
StorageIndex | nc_ |
Definition at line 25 of file TensorContractionBlocking.h.
|
inline |
Definition at line 45 of file TensorContractionBlocking.h.
|
inline |
Definition at line 60 of file TensorContractionBlocking.h.
|
inline |
Definition at line 61 of file TensorContractionBlocking.h.
|
inline |
Definition at line 62 of file TensorContractionBlocking.h.
|
private |
Definition at line 65 of file TensorContractionBlocking.h.
|
private |
Definition at line 66 of file TensorContractionBlocking.h.
|
private |
Definition at line 67 of file TensorContractionBlocking.h.