A cost model used to limit the number of threads used for evaluating tensor expression. More...
#include <TensorEvaluator.h>
Public Types | |
enum | { IsAligned = Derived::IsAligned, PacketAccess = (internal::unpacket_traits<PacketReturnType>::size > 1), Layout = Derived::Layout } |
typedef Derived::Scalar | CoeffReturnType |
typedef Derived::Dimensions | Dimensions |
typedef Derived::Index | Index |
typedef PacketType< CoeffReturnType, Device >::type | PacketReturnType |
typedef Derived::Scalar | Scalar |
Public Member Functions | |
EIGEN_DEVICE_FUNC EIGEN_STRONG_INLINE void | cleanup () |
EIGEN_DEVICE_FUNC EIGEN_STRONG_INLINE CoeffReturnType | coeff (Index index) const |
EIGEN_DEVICE_FUNC EIGEN_STRONG_INLINE CoeffReturnType | coeff (const array< DenseIndex, NumCoords > &coords) const |
EIGEN_DEVICE_FUNC EIGEN_STRONG_INLINE Scalar & | coeffRef (Index index) |
EIGEN_DEVICE_FUNC EIGEN_STRONG_INLINE Scalar & | coeffRef (const array< DenseIndex, NumCoords > &coords) |
EIGEN_DEVICE_FUNC EIGEN_STRONG_INLINE TensorOpCost | costPerCoeff (bool vectorized) const |
EIGEN_DEVICE_FUNC internal::traits< Derived >::template MakePointer< Scalar >::Type | data () const |
const Derived & | derived () const |
const Device & | device () const |
required by sycl in order to construct sycl buffer from raw pointer More... | |
EIGEN_DEVICE_FUNC EIGEN_STRONG_INLINE const Dimensions & | dimensions () const |
EIGEN_DEVICE_FUNC EIGEN_STRONG_INLINE bool | evalSubExprsIfNeeded (CoeffReturnType *dest) |
template<int LoadMode> | |
EIGEN_DEVICE_FUNC EIGEN_STRONG_INLINE PacketReturnType | packet (Index index) const |
EIGEN_DEVICE_FUNC EIGEN_STRONG_INLINE | TensorEvaluator (const Derived &m, const Device &device) |
template<int StoreMode> | |
EIGEN_DEVICE_FUNC EIGEN_STRONG_INLINE void | writePacket (Index index, const PacketReturnType &x) |
Static Public Attributes | |
static const int | NumCoords |
Protected Attributes | |
internal::traits< Derived >::template MakePointer< Scalar >::Type | m_data |
const Device & | m_device |
Dimensions | m_dims |
const Derived & | m_impl |
A cost model used to limit the number of threads used for evaluating tensor expression.
The tensor evaluator classes.
These classes are responsible for the evaluation of the tensor expression.
TODO: add support for more types of expressions, in particular expressions leading to lvalues (slicing, reshaping, etc...)
Definition at line 28 of file TensorEvaluator.h.
typedef Derived::Scalar Eigen::TensorEvaluator< Derived, Device >::CoeffReturnType |
Definition at line 32 of file TensorEvaluator.h.
typedef Derived::Dimensions Eigen::TensorEvaluator< Derived, Device >::Dimensions |
Definition at line 34 of file TensorEvaluator.h.
typedef Derived::Index Eigen::TensorEvaluator< Derived, Device >::Index |
Definition at line 30 of file TensorEvaluator.h.
typedef PacketType<CoeffReturnType, Device>::type Eigen::TensorEvaluator< Derived, Device >::PacketReturnType |
Definition at line 33 of file TensorEvaluator.h.
typedef Derived::Scalar Eigen::TensorEvaluator< Derived, Device >::Scalar |
Definition at line 31 of file TensorEvaluator.h.
anonymous enum |
Enumerator | |
---|---|
IsAligned | |
PacketAccess | |
Layout |
Definition at line 40 of file TensorEvaluator.h.
|
inline |
Definition at line 48 of file TensorEvaluator.h.
|
inline |
Definition at line 64 of file TensorEvaluator.h.
|
inline |
Definition at line 66 of file TensorEvaluator.h.
|
inline |
Definition at line 88 of file TensorEvaluator.h.
|
inline |
Definition at line 71 of file TensorEvaluator.h.
|
inline |
Definition at line 97 of file TensorEvaluator.h.
|
inline |
Definition at line 106 of file TensorEvaluator.h.
|
inline |
Definition at line 111 of file TensorEvaluator.h.
|
inline |
Definition at line 53 of file TensorEvaluator.h.
|
inline |
required by sycl in order to construct sycl buffer from raw pointer
Definition at line 114 of file TensorEvaluator.h.
|
inline |
Definition at line 54 of file TensorEvaluator.h.
|
inline |
Definition at line 56 of file TensorEvaluator.h.
|
inline |
Definition at line 77 of file TensorEvaluator.h.
|
inline |
Definition at line 83 of file TensorEvaluator.h.
|
protected |
Definition at line 117 of file TensorEvaluator.h.
|
protected |
Definition at line 119 of file TensorEvaluator.h.
|
protected |
Definition at line 118 of file TensorEvaluator.h.
|
protected |
Definition at line 120 of file TensorEvaluator.h.
|
static |
Definition at line 37 of file TensorEvaluator.h.