|
enum | { Layout = TensorEvaluator<LeftArgType, Device>::Layout,
PacketAccess = (PacketType<CoeffReturnType, Device>::size > 1),
BlockAccess = false
} |
|
typedef TensorContractionEvaluatorBase< Self > | Base |
|
typedef XprType::CoeffReturnType | CoeffReturnType |
|
typedef array< StorageIndex, ContractDims > | contract_t |
|
typedef Eigen::SyclDevice | Device |
|
typedef DSizes< StorageIndex, NumDims > | Dimensions |
|
typedef Base::EvaluatorPointerType | EvaluatorPointerType |
|
typedef array< StorageIndex, LDims > | left_dim_mapper_t |
|
typedef array< StorageIndex, LDims - ContractDims > | left_nocontract_t |
|
typedef LeftEvaluator::Dimensions | LeftDimensions |
|
typedef TensorEvaluator< typename Base::EvalLeftArgType, Device > | LeftEvaluator |
|
typedef Eigen::internal::remove_const< typename LeftEvaluator::CoeffReturnType >::type | LhsScalar |
|
typedef PacketType< CoeffReturnType, Device >::type | PacketReturnType |
|
typedef Eigen::internal::remove_const< typename RightEvaluator::CoeffReturnType >::type | RhsScalar |
|
typedef array< StorageIndex, RDims > | right_dim_mapper_t |
|
typedef array< StorageIndex, RDims - ContractDims > | right_nocontract_t |
|
typedef RightEvaluator::Dimensions | RightDimensions |
|
typedef TensorEvaluator< typename Base::EvalRightArgType, Device > | RightEvaluator |
|
typedef internal::remove_const< typename XprType::Scalar >::type | Scalar |
|
typedef TensorEvaluator< const TensorContractionOp< Indices, LeftArgType, RightArgType, OutputKernelType >, Device > | Self |
|
typedef Base::Storage | Storage |
|
typedef XprType::Index | StorageIndex |
|
typedef TensorContractionOp< Indices, LeftArgType, RightArgType, OutputKernelType > | XprType |
|
enum | |
|
typedef XprType::CoeffReturnType | CoeffReturnType |
|
typedef array< Index, ContractDims > | contract_t |
|
typedef internal::traits< TensorEvaluator< const TensorContractionOp< Indices, LeftArgType, RightArgType, OutputKernelType >, Eigen::SyclDevice > >::Device | Device |
|
typedef DSizes< Index, NumDims > | Dimensions |
|
typedef internal::conditional< static_cast< int >Layout)==static_cast< int >ColMajor), LeftArgType, RightArgType >::type | EvalLeftArgType |
|
typedef internal::conditional< static_cast< int >Layout)==static_cast< int >ColMajor), RightArgType, LeftArgType >::type | EvalRightArgType |
|
typedef Storage::Type | EvaluatorPointerType |
|
typedef XprType::Index | Index |
|
typedef internal::traits< TensorEvaluator< const TensorContractionOp< Indices, LeftArgType, RightArgType, OutputKernelType >, Eigen::SyclDevice > >::Indices | Indices |
|
typedef array< Index, LDims - ContractDims > | left_nocontract_t |
|
typedef internal::traits< TensorEvaluator< const TensorContractionOp< Indices, LeftArgType, RightArgType, OutputKernelType >, Eigen::SyclDevice > >::LeftArgType | LeftArgType |
|
typedef TensorEvaluator< EvalLeftArgType, Device > | LeftEvaluatorType |
|
typedef internal::traits< TensorEvaluator< const TensorContractionOp< Indices, LeftArgType, RightArgType, OutputKernelType >, Eigen::SyclDevice > >::OutputKernelType | OutputKernelType |
|
typedef PacketType< CoeffReturnType, Device >::type | PacketReturnType |
|
typedef array< Index, RDims - ContractDims > | right_nocontract_t |
|
typedef internal::traits< TensorEvaluator< const TensorContractionOp< Indices, LeftArgType, RightArgType, OutputKernelType >, Eigen::SyclDevice > >::RightArgType | RightArgType |
|
typedef TensorEvaluator< EvalRightArgType, Device > | RightEvaluatorType |
|
typedef internal::remove_const< typename XprType::Scalar >::type | Scalar |
|
typedef StorageMemory< Scalar, Device > | Storage |
|
typedef internal::TensorBlockNotImplemented | TensorBlock |
|
typedef TensorContractionOp< Indices, LeftArgType, RightArgType, OutputKernelType > | XprType |
|
|
template<bool skinny, typename input_mapper_properties , typename LhsMapper , typename RhsMapper > |
void EIGEN_ALWAYS_INLINE | adjustTT (EvaluatorPointerType buffer, const LhsMapper &lhs, const RhsMapper &rhs, const TripleDim &triple_dim) const |
|
EIGEN_DEVICE_FUNC EIGEN_STRONG_INLINE void | bind (cl::sycl::handler &cgh) const |
|
EIGEN_STRONG_INLINE void | cleanup () |
|
const Eigen::SyclDevice & | device () const |
|
EIGEN_STRONG_INLINE bool | evalSubExprsIfNeeded (typename Base::EvaluatorPointerType data) |
|
void | evalToSycl (typename Base::EvaluatorPointerType buffer) const |
|
template<bool lhs_inner_dim_contiguous, bool rhs_inner_dim_contiguous, bool rhs_inner_dim_reordered, int Alignment> |
void | evalTyped (typename Base::EvaluatorPointerType buffer) const |
|
template<typename LhsMapper , typename RhsMapper > |
EIGEN_ALWAYS_INLINE void | launchSC (EvaluatorPointerType buffer, const LhsMapper &lhs, const RhsMapper &rhs, StorageIndex K) const |
|
template<TensorSycl::internal::contraction_type ct, bool skinny, typename input_mapper_properties , typename Properties , typename LhsMapper , typename RhsMapper > |
void | launchTT (EvaluatorPointerType buffer, const LhsMapper &lhs, const RhsMapper &rhs, const TripleDim &triple_dim) const |
|
template<bool is_lhs_vec, typename VectorMapper , typename TensorMapper , typename StorageIndex > |
void EIGEN_ALWAYS_INLINE | LaunchVT (EvaluatorPointerType buffer, const VectorMapper &vec, const TensorMapper &mat, StorageIndex NC, StorageIndex C) const |
|
| TensorEvaluator (const XprType &op, const Device &device) |
|
EIGEN_STRONG_INLINE void | cleanup () |
|
EIGEN_DEVICE_FUNC EIGEN_STRONG_INLINE CoeffReturnType | coeff (Index index) const |
|
EIGEN_DEVICE_FUNC EIGEN_STRONG_INLINE TensorOpCost | costPerCoeff (bool) const |
|
EIGEN_DEVICE_FUNC EIGEN_STRONG_INLINE EvaluatorPointerType | data () const |
|
EIGEN_DEVICE_FUNC EIGEN_STRONG_INLINE const Dimensions & | dimensions () const |
|
EIGEN_DEVICE_FUNC void | evalGemm (Scalar *buffer) const |
|
EIGEN_DEVICE_FUNC void | evalGemmPartial (Scalar *buffer, Index k_start, Index k_end, int num_threads) const |
|
EIGEN_DEVICE_FUNC void | evalGemmPartialWithoutOutputKernel (Scalar *buffer, Index k_start, Index k_end, int num_threads) const |
|
EIGEN_DEVICE_FUNC void | evalGemv (Scalar *buffer) const |
|
void | evalProductSequential (Scalar *buffer) const |
|
EIGEN_STRONG_INLINE bool | evalSubExprsIfNeeded (EvaluatorPointerType data) |
|
EIGEN_DEVICE_FUNC void | evalTo (Scalar *buffer) const |
|
EIGEN_DEVICE_FUNC EIGEN_STRONG_INLINE PacketReturnType | packet (Index index) const |
|
EIGEN_STRONG_INLINE | TensorContractionEvaluatorBase (const XprType &op, const Device &device) |
|
template<typename Indices, typename LeftArgType, typename RightArgType, typename OutputKernelType>
struct Eigen::TensorEvaluator< const TensorContractionOp< Indices, LeftArgType, RightArgType, OutputKernelType >, Eigen::SyclDevice >
Definition at line 1283 of file TensorContractionSycl.h.
template<typename Indices , typename LeftArgType , typename RightArgType , typename OutputKernelType >
template<TensorSycl::internal::contraction_type ct, bool skinny, typename input_mapper_properties , typename Properties , typename LhsMapper , typename RhsMapper >