|
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) |
|
template<typename LhsScalar, typename RhsScalar>
struct Eigen::internal::TensorContractionBlockMemAllocator< LhsScalar, RhsScalar >
Definition at line 79 of file TensorContraction.h.