#include <TensorContraction.h>
Public Types | |
enum | { HasBeta = false } |
typedef TensorContractionBlockMemAllocator< LhsScalar, RhsScalar > | BlockMemAllocator |
typedef BlockMemAllocator::BlockMemHandle | BlockMemHandle |
typedef internal::gebp_kernel< LhsScalar, RhsScalar, StorageIndex, OutputMapper, Traits::mr, Traits::nr, false, false > | GebpKernel |
typedef LhsScalar * | LhsBlock |
typedef internal::gemm_pack_lhs< LhsScalar, StorageIndex, typename LhsMapper::SubMapper, Traits::mr, Traits::LhsProgress, typename Traits::LhsPacket4Packing, ColMajor > | LhsPacker |
typedef RhsScalar * | RhsBlock |
typedef internal::gemm_pack_rhs< RhsScalar, StorageIndex, typename RhsMapper::SubMapper, Traits::nr, ColMajor > | RhsPacker |
typedef internal::gebp_traits< LhsScalar, RhsScalar > | Traits |
Public Member Functions | |
template<typename Device > | |
EIGEN_DEVICE_FUNC BlockMemHandle | allocate (Device &d, LhsBlock *lhs_block, RhsBlock *rhs_block) |
template<typename Device > | |
EIGEN_DEVICE_FUNC BlockMemHandle | allocateSlices (Device &d, const StorageIndex num_lhs, const StorageIndex num_rhs, const StorageIndex num_slices, std::vector< LhsBlock > *lhs_blocks, std::vector< RhsBlock > *rhs_blocks) |
EIGEN_DEVICE_FUNC EIGEN_DONT_INLINE void | invoke (const OutputMapper &output_mapper, const LhsBlock &lhsBlock, const RhsBlock &rhsBlock, const StorageIndex rows, const StorageIndex depth, const StorageIndex cols, const ResScalar alpha, const ResScalar beta) |
EIGEN_DEVICE_FUNC EIGEN_DONT_INLINE void | packLhs (LhsBlock *lhsBlock, const typename LhsMapper::SubMapper &data_mapper, const StorageIndex depth, const StorageIndex rows) |
EIGEN_DEVICE_FUNC EIGEN_DONT_INLINE void | packRhs (RhsBlock *rhsBlock, const typename RhsMapper::SubMapper &data_mapper, const StorageIndex depth, const StorageIndex cols) |
EIGEN_DEVICE_FUNC | TensorContractionKernel (StorageIndex m_, StorageIndex k_, StorageIndex n_, StorageIndex bm_, StorageIndex bk_, StorageIndex bn_) |
Static Public Member Functions | |
template<typename Device > | |
static EIGEN_DEVICE_FUNC void | deallocate (Device &d, BlockMemHandle handle) |
Private Attributes | |
const StorageIndex | bk |
const StorageIndex | bm |
const StorageIndex | bn |
const StorageIndex | k |
const StorageIndex | m |
const StorageIndex | n |
Definition at line 182 of file TensorContraction.h.
typedef TensorContractionBlockMemAllocator<LhsScalar, RhsScalar> Eigen::internal::TensorContractionKernel< ResScalar, LhsScalar, RhsScalar, StorageIndex, OutputMapper, LhsMapper, RhsMapper >::BlockMemAllocator |
Definition at line 198 of file TensorContraction.h.
typedef BlockMemAllocator::BlockMemHandle Eigen::internal::TensorContractionKernel< ResScalar, LhsScalar, RhsScalar, StorageIndex, OutputMapper, LhsMapper, RhsMapper >::BlockMemHandle |
Definition at line 199 of file TensorContraction.h.
typedef internal::gebp_kernel<LhsScalar, RhsScalar, StorageIndex, OutputMapper, Traits::mr, Traits::nr, false, false> Eigen::internal::TensorContractionKernel< ResScalar, LhsScalar, RhsScalar, StorageIndex, OutputMapper, LhsMapper, RhsMapper >::GebpKernel |
Definition at line 216 of file TensorContraction.h.
typedef LhsScalar* Eigen::internal::TensorContractionKernel< ResScalar, LhsScalar, RhsScalar, StorageIndex, OutputMapper, LhsMapper, RhsMapper >::LhsBlock |
Definition at line 193 of file TensorContraction.h.
typedef internal::gemm_pack_lhs< LhsScalar, StorageIndex, typename LhsMapper::SubMapper, Traits::mr, Traits::LhsProgress, typename Traits::LhsPacket4Packing, ColMajor> Eigen::internal::TensorContractionKernel< ResScalar, LhsScalar, RhsScalar, StorageIndex, OutputMapper, LhsMapper, RhsMapper >::LhsPacker |
Definition at line 206 of file TensorContraction.h.
typedef RhsScalar* Eigen::internal::TensorContractionKernel< ResScalar, LhsScalar, RhsScalar, StorageIndex, OutputMapper, LhsMapper, RhsMapper >::RhsBlock |
Definition at line 194 of file TensorContraction.h.
typedef internal::gemm_pack_rhs<RhsScalar, StorageIndex, typename RhsMapper::SubMapper, Traits::nr, ColMajor> Eigen::internal::TensorContractionKernel< ResScalar, LhsScalar, RhsScalar, StorageIndex, OutputMapper, LhsMapper, RhsMapper >::RhsPacker |
Definition at line 211 of file TensorContraction.h.
typedef internal::gebp_traits<LhsScalar, RhsScalar> Eigen::internal::TensorContractionKernel< ResScalar, LhsScalar, RhsScalar, StorageIndex, OutputMapper, LhsMapper, RhsMapper >::Traits |
Definition at line 201 of file TensorContraction.h.
anonymous enum |
Enumerator | |
---|---|
HasBeta |
Definition at line 185 of file TensorContraction.h.
|
inline |
Definition at line 188 of file TensorContraction.h.
|
inline |
Definition at line 219 of file TensorContraction.h.
|
inline |
Definition at line 225 of file TensorContraction.h.
|
inlinestatic |
Definition at line 234 of file TensorContraction.h.
|
inline |
Definition at line 251 of file TensorContraction.h.
|
inline |
Definition at line 238 of file TensorContraction.h.
|
inline |
Definition at line 245 of file TensorContraction.h.
|
private |
Definition at line 273 of file TensorContraction.h.
|
private |
Definition at line 272 of file TensorContraction.h.
|
private |
Definition at line 274 of file TensorContraction.h.
|
private |
Definition at line 270 of file TensorContraction.h.
|
private |
Definition at line 269 of file TensorContraction.h.
|
private |
Definition at line 271 of file TensorContraction.h.