/external/eigen/unsupported/Eigen/CXX11/src/Tensor/ |
D | TensorSyclExtractFunctors.h | 48 struct FunctorExtractor<TensorEvaluator<const UnaryCategory<OP, RHSExpr>, Dev> > { 49 FunctorExtractor<TensorEvaluator<RHSExpr, Dev> > rhsExpr; 51 FunctorExtractor(const TensorEvaluator<const UnaryCategory<OP, RHSExpr>, Dev>& expr) 57 struct FunctorExtractor<TensorEvaluator<UnaryCategory<OP, RHSExpr>, Dev> > 58 : FunctorExtractor<TensorEvaluator<const UnaryCategory<OP, RHSExpr>, Dev> >{}; 63 struct FunctorExtractor<TensorEvaluator<const BinaryCategory<OP, LHSExpr, RHSExpr>, Dev> > { 64 FunctorExtractor<TensorEvaluator<LHSExpr, Dev> > lhsExpr; 65 FunctorExtractor<TensorEvaluator<RHSExpr, Dev> > rhsExpr; 67 FunctorExtractor(const TensorEvaluator<const BinaryCategory<OP, LHSExpr, RHSExpr>, Dev>& expr) 74 struct FunctorExtractor<TensorEvaluator<BinaryCategory<OP, LHSExpr, RHSExpr>, Dev> > [all …]
|
D | TensorEvaluator.h | 28 struct TensorEvaluator struct 48 EIGEN_DEVICE_FUNC EIGEN_STRONG_INLINE TensorEvaluator(const Derived& m, const Device& device) in TensorEvaluator() argument 148 struct TensorEvaluator<const Derived, Device> 171 EIGEN_DEVICE_FUNC EIGEN_STRONG_INLINE TensorEvaluator(const Derived& m, const Device& device) 228 struct TensorEvaluator<const TensorCwiseNullaryOp<NullaryOp, ArgType>, Device> 235 Layout = TensorEvaluator<ArgType, Device>::Layout, 241 TensorEvaluator(const XprType& op, const Device& device) 250 typedef typename TensorEvaluator<ArgType, Device>::Dimensions Dimensions; 277 const TensorEvaluator<ArgType, Device>& impl() const { return m_argImpl; } 284 TensorEvaluator<ArgType, Device> m_argImpl; [all …]
|
D | TensorSyclExtractAccessor.h | 69 struct ExtractAccessor<TensorEvaluator<const UnaryCategory<OP, RHSExpr>, Dev> > { 70 …static inline auto getTuple(cl::sycl::handler& cgh, const TensorEvaluator<const UnaryCategory<OP, … 78 struct ExtractAccessor<TensorEvaluator<UnaryCategory<OP, RHSExpr>, Dev> > 79 : ExtractAccessor<TensorEvaluator<const UnaryCategory<OP, RHSExpr>, Dev> > {}; 83 struct ExtractAccessor<TensorEvaluator<const BinaryCategory<OP, LHSExpr, RHSExpr>, Dev> > { 84 …static inline auto getTuple(cl::sycl::handler& cgh, const TensorEvaluator<const BinaryCategory<OP,… 91 struct ExtractAccessor<TensorEvaluator<BinaryCategory<OP, LHSExpr, RHSExpr>, Dev> > 92 : ExtractAccessor<TensorEvaluator<const BinaryCategory<OP, LHSExpr, RHSExpr>, Dev> >{}; 97 struct ExtractAccessor<TensorEvaluator<const TernaryCategory<OP, Arg1Expr, Arg2Expr, Arg3Expr>, Dev… 98 …static inline auto getTuple(cl::sycl::handler& cgh, const TensorEvaluator<const TernaryCategory<OP… [all …]
|
D | TensorConversion.h | 52 template <typename TensorEvaluator, typename SrcPacket, typename TgtPacket, int SrcCoeffRatio, int … 55 PacketConverter(const TensorEvaluator& impl) 64 const TensorEvaluator& m_impl; 68 template <typename TensorEvaluator, typename SrcPacket, typename TgtPacket> 69 struct PacketConverter<TensorEvaluator, SrcPacket, TgtPacket, 2, 1> { 71 PacketConverter(const TensorEvaluator& impl) 85 const TensorEvaluator& m_impl; 88 template <typename TensorEvaluator, typename SrcPacket, typename TgtPacket> 89 struct PacketConverter<TensorEvaluator, SrcPacket, TgtPacket, 4, 1> { 91 PacketConverter(const TensorEvaluator& impl) [all …]
|
D | TensorAssign.h | 89 struct TensorEvaluator<const TensorAssignOp<LeftArgType, RightArgType>, Device> 96 typedef typename TensorEvaluator<RightArgType, Device>::Dimensions Dimensions; 100 …IsAligned = TensorEvaluator<LeftArgType, Device>::IsAligned & TensorEvaluator<RightArgType, Device… 101 …PacketAccess = TensorEvaluator<LeftArgType, Device>::PacketAccess & TensorEvaluator<RightArgType, … 102 Layout = TensorEvaluator<LeftArgType, Device>::Layout, 103 RawAccess = TensorEvaluator<LeftArgType, Device>::RawAccess 106 EIGEN_DEVICE_FUNC TensorEvaluator(const XprType& op, const Device& device) : 110 …GEN_STATIC_ASSERT((static_cast<int>(TensorEvaluator<LeftArgType, Device>::Layout) == static_cast<i… 139 const int LhsStoreMode = TensorEvaluator<LeftArgType, Device>::IsAligned ? Aligned : Unaligned; 140 const int RhsLoadMode = TensorEvaluator<RightArgType, Device>::IsAligned ? Aligned : Unaligned; [all …]
|
D | TensorLayoutSwap.h | 111 struct TensorEvaluator<const TensorLayoutSwapOp<ArgType>, Device> 115 …static const int NumDims = internal::array_size<typename TensorEvaluator<ArgType, Device>::Dimensi… 119 IsAligned = TensorEvaluator<ArgType, Device>::IsAligned, 120 PacketAccess = TensorEvaluator<ArgType, Device>::PacketAccess, 121 …Layout = (static_cast<int>(TensorEvaluator<ArgType, Device>::Layout) == static_cast<int>(ColMajor)… 123 RawAccess = TensorEvaluator<ArgType, Device>::RawAccess 126 EIGEN_DEVICE_FUNC EIGEN_STRONG_INLINE TensorEvaluator(const XprType& op, const Device& device) 164 const TensorEvaluator<ArgType, Device>& impl() const { return m_impl; } 167 TensorEvaluator<ArgType, Device> m_impl; 174 struct TensorEvaluator<TensorLayoutSwapOp<ArgType>, Device> [all …]
|
D | TensorMorphing.h | 99 struct TensorEvaluator<const TensorReshapingOp<NewDimensions, ArgType>, Device> 105 IsAligned = TensorEvaluator<ArgType, Device>::IsAligned, 106 PacketAccess = TensorEvaluator<ArgType, Device>::PacketAccess, 107 Layout = TensorEvaluator<ArgType, Device>::Layout, 109 RawAccess = TensorEvaluator<ArgType, Device>::RawAccess 112 EIGEN_DEVICE_FUNC EIGEN_STRONG_INLINE TensorEvaluator(const XprType& op, const Device& device) 151 EIGEN_DEVICE_FUNC const TensorEvaluator<ArgType, Device>& impl() const { return m_impl; } 154 TensorEvaluator<ArgType, Device> m_impl; 161 struct TensorEvaluator<TensorReshapingOp<NewDimensions, ArgType>, Device> 162 : public TensorEvaluator<const TensorReshapingOp<NewDimensions, ArgType>, Device> [all …]
|
D | TensorEvalTo.h | 93 struct TensorEvaluator<const TensorEvalToOp<ArgType, MakePointer_>, Device> 97 typedef typename TensorEvaluator<ArgType, Device>::Dimensions Dimensions; 104 IsAligned = TensorEvaluator<ArgType, Device>::IsAligned, 105 PacketAccess = TensorEvaluator<ArgType, Device>::PacketAccess, 106 Layout = TensorEvaluator<ArgType, Device>::Layout, 111 EIGEN_DEVICE_FUNC EIGEN_STRONG_INLINE TensorEvaluator(const XprType& op, const Device& device) 121 EIGEN_DEVICE_FUNC EIGEN_STRONG_INLINE ~TensorEvaluator() { 137 …ype, PacketReturnType, Aligned>(m_buffer + i, m_impl.template packet<TensorEvaluator<ArgType, Devi… 166 const TensorEvaluator<ArgType, Device>& impl() const { return m_impl; } 171 TensorEvaluator<ArgType, Device> m_impl;
|
D | TensorConcatenation.h | 110 struct TensorEvaluator<const TensorConcatenationOp<Axis, LeftArgType, RightArgType>, Device> 114 …static const int NumDims = internal::array_size<typename TensorEvaluator<LeftArgType, Device>::Dim… 115 …static const int RightNumDims = internal::array_size<typename TensorEvaluator<RightArgType, Device… 122 …PacketAccess = TensorEvaluator<LeftArgType, Device>::PacketAccess & TensorEvaluator<RightArgType, … 123 Layout = TensorEvaluator<LeftArgType, Device>::Layout, 127 EIGEN_DEVICE_FUNC EIGEN_STRONG_INLINE TensorEvaluator(const XprType& op, const Device& device) 130 …GEN_STATIC_ASSERT((static_cast<int>(TensorEvaluator<LeftArgType, Device>::Layout) == static_cast<i… 285 TensorEvaluator<LeftArgType, Device> m_leftImpl; 286 TensorEvaluator<RightArgType, Device> m_rightImpl; 292 struct TensorEvaluator<TensorConcatenationOp<Axis, LeftArgType, RightArgType>, Device> [all …]
|
D | TensorArgMax.h | 76 struct TensorEvaluator<const TensorIndexTupleOp<ArgType>, Device> 83 typedef typename TensorEvaluator<ArgType, Device>::Dimensions Dimensions; 90 Layout = TensorEvaluator<ArgType, Device>::Layout, 95 EIGEN_DEVICE_FUNC EIGEN_STRONG_INLINE TensorEvaluator(const XprType& op, const Device& device) 123 TensorEvaluator<ArgType, Device> m_impl; 201 struct TensorEvaluator<const TensorTupleReducerOp<ReduceOp, Dims, ArgType>, Device> 208 …typedef typename TensorEvaluator<const TensorReductionOp<ReduceOp, Dims, const TensorIndexTupleOp<… 209 …typedef typename TensorEvaluator<const TensorIndexTupleOp<ArgType> , Device>::Dimensions InputDime… 217 …Layout = TensorEvaluator<const TensorReductionOp<ReduceOp, Dims, const TensorIndexTupleOp<ArgType>… 222 EIGEN_DEVICE_FUNC EIGEN_STRONG_INLINE TensorEvaluator(const XprType& op, const Device& device) [all …]
|
D | TensorConvolution.h | 291 struct TensorEvaluator<const TensorConvolutionOp<Indices, InputArgType, KernelArgType>, Device> 295 …static const int NumDims = internal::array_size<typename TensorEvaluator<InputArgType, Device>::Di… 306 …IsAligned = TensorEvaluator<InputArgType, Device>::IsAligned & TensorEvaluator<KernelArgType, Devi… 307 …PacketAccess = TensorEvaluator<InputArgType, Device>::PacketAccess & TensorEvaluator<KernelArgType… 308 Layout = TensorEvaluator<InputArgType, Device>::Layout, 313 EIGEN_DEVICE_FUNC EIGEN_STRONG_INLINE TensorEvaluator(const XprType& op, const Device& device) 316 …EN_STATIC_ASSERT((static_cast<int>(TensorEvaluator<InputArgType, Device>::Layout) == static_cast<i… 318 …const typename TensorEvaluator<InputArgType, Device>::Dimensions& input_dims = m_inputImpl.dimensi… 319 …const typename TensorEvaluator<KernelArgType, Device>::Dimensions& kernel_dims = m_kernelImpl.dime… 540 TensorEvaluator<InputArgType, Device> m_inputImpl; [all …]
|
D | TensorReverse.h | 101 struct TensorEvaluator<const TensorReverseOp<ReverseDimensions, ArgType>, Device> 114 PacketAccess = TensorEvaluator<ArgType, Device>::PacketAccess, 115 Layout = TensorEvaluator<ArgType, Device>::Layout, 120 EIGEN_DEVICE_FUNC EIGEN_STRONG_INLINE TensorEvaluator(const XprType& op, 230 TensorEvaluator<ArgType, Device> m_impl; 237 struct TensorEvaluator<TensorReverseOp<ReverseDimensions, ArgType>, Device> 238 : public TensorEvaluator<const TensorReverseOp<ReverseDimensions, ArgType>, 240 typedef TensorEvaluator<const TensorReverseOp<ReverseDimensions, ArgType>, 249 PacketAccess = TensorEvaluator<ArgType, Device>::PacketAccess, 250 Layout = TensorEvaluator<ArgType, Device>::Layout, [all …]
|
D | TensorShuffling.h | 100 struct TensorEvaluator<const TensorShufflingOp<Shuffle, ArgType>, Device> 104 …static const int NumDims = internal::array_size<typename TensorEvaluator<ArgType, Device>::Dimensi… 114 Layout = TensorEvaluator<ArgType, Device>::Layout, 119 EIGEN_DEVICE_FUNC EIGEN_STRONG_INLINE TensorEvaluator(const XprType& op, const Device& device) 122 const typename TensorEvaluator<ArgType, Device>::Dimensions& input_dims = m_impl.dimensions(); 213 TensorEvaluator<ArgType, Device> m_impl; 219 struct TensorEvaluator<TensorShufflingOp<Shuffle, ArgType>, Device> 220 : public TensorEvaluator<const TensorShufflingOp<Shuffle, ArgType>, Device> 222 typedef TensorEvaluator<const TensorShufflingOp<Shuffle, ArgType>, Device> Base; 226 …static const int NumDims = internal::array_size<typename TensorEvaluator<ArgType, Device>::Dimensi… [all …]
|
D | TensorStriding.h | 100 struct TensorEvaluator<const TensorStridingOp<Strides, ArgType>, Device> 104 …static const int NumDims = internal::array_size<typename TensorEvaluator<ArgType, Device>::Dimensi… 113 PacketAccess = TensorEvaluator<ArgType, Device>::PacketAccess, 114 Layout = TensorEvaluator<ArgType, Device>::Layout, 119 EIGEN_DEVICE_FUNC EIGEN_STRONG_INLINE TensorEvaluator(const XprType& op, const Device& device) 127 const typename TensorEvaluator<ArgType, Device>::Dimensions& input_dims = m_impl.dimensions(); 252 TensorEvaluator<ArgType, Device> m_impl; 258 struct TensorEvaluator<TensorStridingOp<Strides, ArgType>, Device> 259 : public TensorEvaluator<const TensorStridingOp<Strides, ArgType>, Device> 262 typedef TensorEvaluator<const XprType, Device> Base; [all …]
|
D | TensorContraction.h | 62 struct traits<TensorEvaluator<const TensorContractionOp<Indices_, LeftArgType_, RightArgType_>, Dev… 125 Layout = TensorEvaluator<LeftArgType, Device>::Layout, 140 internal::array_size<typename TensorEvaluator<EvalLeftArgType, Device>::Dimensions>::value; 142 internal::array_size<typename TensorEvaluator<EvalRightArgType, Device>::Dimensions>::value; 160 EIGEN_STATIC_ASSERT((static_cast<int>(TensorEvaluator<LeftArgType, Device>::Layout) == 161 static_cast<int>(TensorEvaluator<RightArgType, Device>::Layout)), 389 typedef TensorEvaluator<EvalLeftArgType, Device> LeftEvaluator; 390 typedef TensorEvaluator<EvalRightArgType, Device> RightEvaluator; 445 typedef TensorEvaluator<EvalLeftArgType, Device> LeftEvaluator; 446 typedef TensorEvaluator<EvalRightArgType, Device> RightEvaluator; [all …]
|
D | TensorChipping.h | 129 struct TensorEvaluator<const TensorChippingOp<DimId, ArgType>, Device> 132 …static const int NumInputDims = internal::array_size<typename TensorEvaluator<ArgType, Device>::Di… 146 PacketAccess = TensorEvaluator<ArgType, Device>::PacketAccess, 147 Layout = TensorEvaluator<ArgType, Device>::Layout, 152 EIGEN_DEVICE_FUNC EIGEN_STRONG_INLINE TensorEvaluator(const XprType& op, const Device& device) 158 const typename TensorEvaluator<ArgType, Device>::Dimensions& input_dims = m_impl.dimensions(); 304 TensorEvaluator<ArgType, Device> m_impl; 312 struct TensorEvaluator<TensorChippingOp<DimId, ArgType>, Device> 313 : public TensorEvaluator<const TensorChippingOp<DimId, ArgType>, Device> 315 typedef TensorEvaluator<const TensorChippingOp<DimId, ArgType>, Device> Base; [all …]
|
D | TensorInitializer.h | 31 static void run(TensorEvaluator<Derived, DefaultDevice>& tensor, in run() 46 static void run(TensorEvaluator<Derived, DefaultDevice>& tensor, 62 static void run(TensorEvaluator<Derived, DefaultDevice>& tensor, 71 void initialize_tensor(TensorEvaluator<Derived, DefaultDevice>& tensor,
|
D | TensorForcedEval.h | 92 struct TensorEvaluator<const TensorForcedEvalOp<ArgType, MakePointer_>, Device> 96 typedef typename TensorEvaluator<ArgType, Device>::Dimensions Dimensions; 105 Layout = TensorEvaluator<ArgType, Device>::Layout, 109 EIGEN_DEVICE_FUNC TensorEvaluator(const XprType& op, const Device& device) 154 const TensorEvaluator<ArgType, Device>& impl() { return m_impl; } 158 TensorEvaluator<ArgType, Device> m_impl;
|
D | TensorInflation.h | 78 struct TensorEvaluator<const TensorInflationOp<Strides, ArgType>, Device> 82 …static const int NumDims = internal::array_size<typename TensorEvaluator<ArgType, Device>::Dimensi… 91 PacketAccess = TensorEvaluator<ArgType, Device>::PacketAccess, 93 Layout = TensorEvaluator<ArgType, Device>::Layout, 98 EIGEN_DEVICE_FUNC EIGEN_STRONG_INLINE TensorEvaluator(const XprType& op, const Device& device) 112 const typename TensorEvaluator<ArgType, Device>::Dimensions& input_dims = m_impl.dimensions(); 222 TensorEvaluator<ArgType, Device> m_impl;
|
D | TensorRef.h | 43 …luatorReadOnly : public TensorLazyBaseEvaluator<Dimensions, typename TensorEvaluator<Expr, Device>… 46 typedef typename TensorEvaluator<Expr, Device>::Scalar Scalar; 72 TensorEvaluator<Expr, Device> m_impl; 356 struct TensorEvaluator<const TensorRef<Derived>, Device> 372 EIGEN_DEVICE_FUNC EIGEN_STRONG_INLINE TensorEvaluator(const TensorRef<Derived>& m, const Device&) 401 struct TensorEvaluator<TensorRef<Derived>, Device> : public TensorEvaluator<const TensorRef<Derived… 409 typedef TensorEvaluator<const TensorRef<Derived>, Device> Base; 417 …EIGEN_DEVICE_FUNC EIGEN_STRONG_INLINE TensorEvaluator(TensorRef<Derived>& m, const Device& d) : Ba…
|
D | TensorScan.h | 84 struct TensorEvaluator<const TensorScanOp<Op, ArgType>, Device> { 88 …static const int NumDims = internal::array_size<typename TensorEvaluator<ArgType, Device>::Dimensi… 93 typedef TensorEvaluator<const TensorScanOp<Op, ArgType>, Device> Self; 99 Layout = TensorEvaluator<ArgType, Device>::Layout, 104 EIGEN_DEVICE_FUNC EIGEN_STRONG_INLINE TensorEvaluator(const XprType& op, 151 EIGEN_DEVICE_FUNC EIGEN_STRONG_INLINE const TensorEvaluator<ArgType, Device>& inner() const { 201 TensorEvaluator<ArgType, Device> m_impl;
|
D | TensorPatch.h | 81 struct TensorEvaluator<const TensorPatchOp<PatchDim, ArgType>, Device> 85 …static const int NumDims = internal::array_size<typename TensorEvaluator<ArgType, Device>::Dimensi… 95 PacketAccess = TensorEvaluator<ArgType, Device>::PacketAccess, 96 Layout = TensorEvaluator<ArgType, Device>::Layout, 101 EIGEN_DEVICE_FUNC EIGEN_STRONG_INLINE TensorEvaluator(const XprType& op, const Device& device) 105 const typename TensorEvaluator<ArgType, Device>::Dimensions& input_dims = m_impl.dimensions(); 264 TensorEvaluator<ArgType, Device> m_impl;
|
D | TensorGenerator.h | 81 struct TensorEvaluator<const TensorGeneratorOp<Generator, ArgType>, Device> 85 typedef typename TensorEvaluator<ArgType, Device>::Dimensions Dimensions; 94 Layout = TensorEvaluator<ArgType, Device>::Layout, 99 EIGEN_DEVICE_FUNC EIGEN_STRONG_INLINE TensorEvaluator(const XprType& op, const Device& device) 102 TensorEvaluator<ArgType, Device> impl(op.expression(), device);
|
D | TensorBroadcasting.h | 96 struct TensorEvaluator<const TensorBroadcastingOp<Broadcast, ArgType>, Device> 100 …static const int NumDims = internal::array_size<typename TensorEvaluator<ArgType, Device>::Dimensi… 103 typedef typename TensorEvaluator<ArgType, Device>::Dimensions InputDimensions; 110 PacketAccess = TensorEvaluator<ArgType, Device>::PacketAccess, 111 Layout = TensorEvaluator<ArgType, Device>::Layout, 115 EIGEN_DEVICE_FUNC EIGEN_STRONG_INLINE TensorEvaluator(const XprType& op, const Device& device) 377 const TensorEvaluator<ArgType, Device>& impl() const { return m_impl; } 386 TensorEvaluator<ArgType, Device> m_impl;
|
/external/eigen/unsupported/test/ |
D | cxx11_tensor_chipping.cpp | 348 typedef TensorEvaluator<decltype(tensor.chip<4>(3)), DefaultDevice> Evaluator4; in test_chip_raw_data_col_major() 361 typedef TensorEvaluator<decltype(tensor.chip<0>(0)), DefaultDevice> Evaluator0; in test_chip_raw_data_col_major() 365 typedef TensorEvaluator<decltype(tensor.chip<1>(0)), DefaultDevice> Evaluator1; in test_chip_raw_data_col_major() 369 typedef TensorEvaluator<decltype(tensor.chip<2>(0)), DefaultDevice> Evaluator2; in test_chip_raw_data_col_major() 373 typedef TensorEvaluator<decltype(tensor.chip<3>(0)), DefaultDevice> Evaluator3; in test_chip_raw_data_col_major() 383 typedef TensorEvaluator<decltype(tensor.chip<0>(3)), DefaultDevice> Evaluator0; in test_chip_raw_data_row_major() 396 typedef TensorEvaluator<decltype(tensor.chip<1>(0)), DefaultDevice> Evaluator1; in test_chip_raw_data_row_major() 400 typedef TensorEvaluator<decltype(tensor.chip<2>(0)), DefaultDevice> Evaluator2; in test_chip_raw_data_row_major() 404 typedef TensorEvaluator<decltype(tensor.chip<3>(0)), DefaultDevice> Evaluator3; in test_chip_raw_data_row_major() 408 typedef TensorEvaluator<decltype(tensor.chip<4>(0)), DefaultDevice> Evaluator4; in test_chip_raw_data_row_major()
|