#include <TensorContraction.h>
Classes | |
struct | BlockSizes |
Public Types | |
typedef void * | BlockMemHandle |
Static Public Member Functions | |
template<typename Device > | |
static EIGEN_DEVICE_FUNC BlockMemHandle | allocate (Device &d, const Index bm, const Index bk, const Index bn, LhsScalar **lhs_block, RhsScalar **rhs_block) |
template<typename Device > | |
static EIGEN_DEVICE_FUNC BlockMemHandle | allocateSlices (Device &d, const Index bm, const Index bk, const Index bn, const Index num_lhs, const Index num_rhs, const Index num_slices, std::vector< LhsScalar * > *lhs_blocks, std::vector< RhsScalar * > *rhs_blocks) |
template<typename Device > | |
static EIGEN_DEVICE_FUNC void | deallocate (Device &d, BlockMemHandle handle) |
Static Private Member Functions | |
static EIGEN_DEVICE_FUNC BlockSizes | ComputeLhsRhsBlockSizes (const Index bm, const Index bk, const Index bn) |
Definition at line 79 of file TensorContraction.h.
typedef void* Eigen::internal::TensorContractionBlockMemAllocator< LhsScalar, RhsScalar >::BlockMemHandle |
Definition at line 80 of file TensorContraction.h.
|
inlinestatic |
Definition at line 83 of file TensorContraction.h.
|
inlinestatic |
Definition at line 99 of file TensorContraction.h.
|
inlinestaticprivate |
Definition at line 140 of file TensorContraction.h.
|
inlinestatic |
Definition at line 131 of file TensorContraction.h.