#include <TensorBroadcasting.h>
Classes | |
struct | BlockBroadcastingIteratorState |
struct | BlockBroadcastingParams |
Static Public Attributes | |
static const int | NumDims = internal::array_size<typename TensorEvaluator<ArgType, Device>::Dimensions>::value |
static const int | PacketSize = PacketType<CoeffReturnType, Device>::size |
Protected Attributes | |
bool | isCopy |
const internal::remove_reference< Broadcast >::type | m_broadcast |
const Device EIGEN_DEVICE_REF | m_device |
Dimensions | m_dimensions |
TensorEvaluator< ArgType, Device > | m_impl |
array< Index, NumDims > | m_inputStrides |
array< Index, NumDims > | m_outputStrides |
bool | nByOne |
bool | oneByN |
Private Member Functions | |
EIGEN_DEVICE_FUNC EIGEN_ALWAYS_INLINE BlockBroadcastingParams | blockBroadcastingParams (TensorBlockDesc &desc) const |
EIGEN_DEVICE_FUNC EIGEN_STRONG_INLINE Index | BroadcastBlock (const Dimensions &input_block_sizes, const Dimensions &input_block_strides, const BroadcastDimensions &bcast_block_sizes, const BroadcastDimensions &bcast_block_strides, const BroadcastDimensions &bcast_input_strides, Index bcast_offset, Index offset, TensorBlockScratch &scratch, ScalarNoConst *materialized_output, ScalarNoConst **materialized_input, size_t *materialized_input_size) const |
EIGEN_DEVICE_FUNC EIGEN_STRONG_INLINE Index | BroadcastBlockAlongBcastDim (BlockBroadcastingParams params, Index bcast_offset, TensorBlockScratch &scratch, ScalarNoConst *materialized_output, ScalarNoConst **materialized_input, size_t *materialized_input_size) const |
EIGEN_DEVICE_FUNC EIGEN_STRONG_INLINE TensorBlock | emptyBlock () const |
Static Private Attributes | |
static const bool | IsColMajor |
Definition at line 97 of file TensorBroadcasting.h.
typedef TensorEvaluator<const ArgType, Device>::TensorBlock Eigen::TensorEvaluator< const TensorBroadcastingOp< Broadcast, ArgType >, Device >::ArgTensorBlock |
Definition at line 134 of file TensorBroadcasting.h.
typedef DSizes<Index, 2 * NumDims> Eigen::TensorEvaluator< const TensorBroadcastingOp< Broadcast, ArgType >, Device >::BroadcastDimensions |
Definition at line 127 of file TensorBroadcasting.h.
typedef XprType::CoeffReturnType Eigen::TensorEvaluator< const TensorBroadcastingOp< Broadcast, ArgType >, Device >::CoeffReturnType |
Definition at line 105 of file TensorBroadcasting.h.
typedef DSizes<Index, NumDims> Eigen::TensorEvaluator< const TensorBroadcastingOp< Broadcast, ArgType >, Device >::Dimensions |
Definition at line 102 of file TensorBroadcasting.h.
typedef Storage::Type Eigen::TensorEvaluator< const TensorBroadcastingOp< Broadcast, ArgType >, Device >::EvaluatorPointerType |
Definition at line 112 of file TensorBroadcasting.h.
typedef XprType::Index Eigen::TensorEvaluator< const TensorBroadcastingOp< Broadcast, ArgType >, Device >::Index |
Definition at line 100 of file TensorBroadcasting.h.
typedef TensorEvaluator<ArgType, Device>::Dimensions Eigen::TensorEvaluator< const TensorBroadcastingOp< Broadcast, ArgType >, Device >::InputDimensions |
Definition at line 104 of file TensorBroadcasting.h.
typedef PacketType<CoeffReturnType, Device>::type Eigen::TensorEvaluator< const TensorBroadcastingOp< Broadcast, ArgType >, Device >::PacketReturnType |
Definition at line 106 of file TensorBroadcasting.h.
typedef XprType::Scalar Eigen::TensorEvaluator< const TensorBroadcastingOp< Broadcast, ArgType >, Device >::Scalar |
Definition at line 103 of file TensorBroadcasting.h.
typedef internal::remove_const<Scalar>::type Eigen::TensorEvaluator< const TensorBroadcastingOp< Broadcast, ArgType >, Device >::ScalarNoConst |
Definition at line 123 of file TensorBroadcasting.h.
typedef StorageMemory<CoeffReturnType, Device> Eigen::TensorEvaluator< const TensorBroadcastingOp< Broadcast, ArgType >, Device >::Storage |
Definition at line 111 of file TensorBroadcasting.h.
typedef internal::TensorMaterializedBlock<ScalarNoConst, NumDims, Layout, Index> Eigen::TensorEvaluator< const TensorBroadcastingOp< Broadcast, ArgType >, Device >::TensorBlock |
Definition at line 138 of file TensorBroadcasting.h.
typedef internal::TensorBlockDescriptor<NumDims, Index> Eigen::TensorEvaluator< const TensorBroadcastingOp< Broadcast, ArgType >, Device >::TensorBlockDesc |
Definition at line 130 of file TensorBroadcasting.h.
typedef internal::TensorBlockScratchAllocator<Device> Eigen::TensorEvaluator< const TensorBroadcastingOp< Broadcast, ArgType >, Device >::TensorBlockScratch |
Definition at line 131 of file TensorBroadcasting.h.
typedef TensorBroadcastingOp<Broadcast, ArgType> Eigen::TensorEvaluator< const TensorBroadcastingOp< Broadcast, ArgType >, Device >::XprType |
Definition at line 99 of file TensorBroadcasting.h.
anonymous enum |
Enumerator | |
---|---|
IsAligned | |
PacketAccess | |
BlockAccess | |
PreferBlockAccess | |
Layout | |
RawAccess |
Definition at line 114 of file TensorBroadcasting.h.
|
inline |
Definition at line 141 of file TensorBroadcasting.h.
|
inline |
Definition at line 629 of file TensorBroadcasting.h.
|
inlineprivate |
Definition at line 753 of file TensorBroadcasting.h.
|
inlineprivate |
Definition at line 1016 of file TensorBroadcasting.h.
|
inlineprivate |
Definition at line 849 of file TensorBroadcasting.h.
|
inline |
Definition at line 226 of file TensorBroadcasting.h.
|
inline |
Definition at line 230 of file TensorBroadcasting.h.
|
inline |
Definition at line 282 of file TensorBroadcasting.h.
|
inline |
Definition at line 317 of file TensorBroadcasting.h.
|
inline |
Definition at line 594 of file TensorBroadcasting.h.
|
inline |
Definition at line 691 of file TensorBroadcasting.h.
|
inline |
Definition at line 211 of file TensorBroadcasting.h.
|
inlineprivate |
Definition at line 843 of file TensorBroadcasting.h.
|
inline |
Definition at line 213 of file TensorBroadcasting.h.
|
inline |
Definition at line 695 of file TensorBroadcasting.h.
|
inline |
Definition at line 619 of file TensorBroadcasting.h.
|
inline |
Definition at line 693 of file TensorBroadcasting.h.
|
inline |
Definition at line 252 of file TensorBroadcasting.h.
|
inline |
Definition at line 287 of file TensorBroadcasting.h.
|
inline |
Definition at line 323 of file TensorBroadcasting.h.
|
inline |
Definition at line 478 of file TensorBroadcasting.h.
|
inline |
Definition at line 441 of file TensorBroadcasting.h.
|
inline |
Definition at line 411 of file TensorBroadcasting.h.
|
inline |
Definition at line 369 of file TensorBroadcasting.h.
|
inline |
Definition at line 536 of file TensorBroadcasting.h.
|
staticprivate |
Definition at line 704 of file TensorBroadcasting.h.
|
protected |
Definition at line 109 of file TensorBroadcasting.h.
|
protected |
Definition at line 1083 of file TensorBroadcasting.h.
|
protected |
Definition at line 1082 of file TensorBroadcasting.h.
|
protected |
Definition at line 1084 of file TensorBroadcasting.h.
|
protected |
Definition at line 1087 of file TensorBroadcasting.h.
|
protected |
Definition at line 1086 of file TensorBroadcasting.h.
|
protected |
Definition at line 1085 of file TensorBroadcasting.h.
|
protected |
Definition at line 109 of file TensorBroadcasting.h.
|
static |
Definition at line 101 of file TensorBroadcasting.h.
|
protected |
Definition at line 109 of file TensorBroadcasting.h.
|
static |
Definition at line 107 of file TensorBroadcasting.h.