TensorBlock.h File Reference

Go to the source code of this file.

Classes
struct	Eigen::internal::TensorBlockScratchAllocator< Device >::Allocation
struct	Eigen::internal::TensorBlockAssignment< Scalar, NumDims, TensorBlockExpr, IndexType >::BlockIteratorState
struct	Eigen::internal::TensorBlockIO< Scalar, IndexType, NumDims, Layout >::BlockIteratorState
class	Eigen::internal::TensorBlockDescriptor< NumDims, IndexType >::DestinationBuffer
struct	Eigen::internal::TensorBlockIO< Scalar, IndexType, NumDims, Layout >::Dst
struct	Eigen::internal::StridedLinearBufferCopy< Scalar, IndexType >::Dst
struct	Eigen::internal::TensorBlockAssignment< Scalar, NumDims, TensorBlockExpr, IndexType >::InnerDimAssign< Vectorizable, Evaluator >
struct	Eigen::internal::TensorBlockAssignment< Scalar, NumDims, TensorBlockExpr, IndexType >::InnerDimAssign< true, Evaluator >
struct	Eigen::internal::TensorBlockIO< Scalar, IndexType, NumDims, Layout >::Src
struct	Eigen::internal::StridedLinearBufferCopy< Scalar, IndexType >::Src
class	Eigen::internal::TensorMaterializedBlock< Scalar, NumDims, Layout, IndexType >::Storage
class	Eigen::internal::StridedLinearBufferCopy< Scalar, IndexType >
struct	Eigen::internal::TensorBlockAssignment< Scalar, NumDims, TensorBlockExpr, IndexType >::Target
class	Eigen::internal::TensorBlockAssignment< Scalar, NumDims, TensorBlockExpr, IndexType >
class	Eigen::internal::TensorBlockDescriptor< NumDims, IndexType >
class	Eigen::internal::TensorBlockIO< Scalar, IndexType, NumDims, Layout >
class	Eigen::internal::TensorBlockIO< Scalar, IndexType, NumDims, Layout >
class	Eigen::internal::TensorBlockMapper< NumDims, Layout, IndexType >
class	Eigen::internal::TensorBlockNotImplemented
struct	Eigen::internal::TensorBlockResourceRequirements
class	Eigen::internal::TensorBlockScratchAllocator< Device >
class	Eigen::internal::TensorCwiseBinaryBlock< BinaryOp, LhsTensorBlock, RhsTensorBlock >
class	Eigen::internal::TensorCwiseUnaryBlock< UnaryOp, ArgTensorBlock >
class	Eigen::internal::TensorMaterializedBlock< Scalar, NumDims, Layout, IndexType >
class	Eigen::internal::TensorTernaryExprBlock< BlockFactory, Arg1TensorBlock, Arg2TensorBlock, Arg3TensorBlock >
class	Eigen::internal::TensorUnaryExprBlock< BlockFactory, ArgTensorBlock >
struct	Eigen::internal::XprScalar< XprType >
struct	Eigen::internal::XprScalar< void >

Namespaces
	Eigen
	Namespace containing all symbols from the Eigen library.
	Eigen::internal

Macros
#define	COPY_INNER_DIM(KIND)

Enumerations
enum	Eigen::internal::TensorBlockKind { Eigen::internal::kExpr, Eigen::internal::kView, Eigen::internal::kMaterializedInScratch, Eigen::internal::kMaterializedInOutput }
enum	Eigen::internal::TensorBlockShapeType { Eigen::internal::TensorBlockShapeType::kUniformAllDims, Eigen::internal::TensorBlockShapeType::kSkewedInnerDims }

Functions
template<int Layout, typename IndexType , int NumDims>
EIGEN_ALWAYS_INLINE DSizes< IndexType, NumDims >	Eigen::internal::strides (const DSizes< IndexType, NumDims > &dimensions)
template<int Layout, typename IndexType , size_t NumDims>
EIGEN_ALWAYS_INLINE DSizes< IndexType, NumDims >	Eigen::internal::strides (const Eigen::array< IndexType, NumDims > &dimensions)
template<int Layout, std::ptrdiff_t... Indices>
EIGEN_STRONG_INLINE DSizes< std::ptrdiff_t, sizeof...(Indices)>	Eigen::internal::strides (const Sizes< Indices... > &sizes)

Macro Definition Documentation

COPY_INNER_DIM

#define COPY_INNER_DIM

(

KIND

)

Value:

IndexType num_copied = 0;                                            \
  for (num_copied = 0; num_copied < block_total_size;                  \
       num_copied += dst_inner_dim_size) {                             \
    LinCopy::template Run<KIND>(                                       \
        typename LinCopy::Dst(output_offset, output_stride, dst.data), \
        typename LinCopy::Src(input_offset, input_stride, src.data),   \
        dst_inner_dim_size);                                           \
                                                                       \
    for (int j = 0; j < idx; ++j) {                                    \
      if (++it[j].count < it[j].size) {                                \
        input_offset += it[j].input_stride;                            \
        output_offset += it[j].output_stride;                          \
        break;                                                         \
      }                                                                \
      it[j].count = 0;                                                 \
      input_offset -= it[j].input_span;                                \
      output_offset -= it[j].output_span;                              \
    }                                                                  \
  }                                                                    \
  return num_copied;