Home
last modified time | relevance | path

Searched refs:TensorEvaluator (Results 1 – 25 of 51) sorted by relevance

123

/external/eigen/unsupported/Eigen/CXX11/src/Tensor/
DTensorSyclExtractFunctors.h48 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 …]
DTensorEvaluator.h28 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 …]
DTensorSyclExtractAccessor.h69 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 …]
DTensorConversion.h52 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 …]
DTensorAssign.h89 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 …]
DTensorLayoutSwap.h111 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 …]
DTensorEvalTo.h93 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;
DTensorMorphing.h99 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 …]
DTensorConcatenation.h110 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 …]
DTensorArgMax.h76 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 …]
DTensorShuffling.h100 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 …]
DTensorReverse.h101 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 …]
DTensorStriding.h100 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 …]
DTensorConvolution.h291 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 …]
DTensorChipping.h129 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 …]
DTensorContraction.h62 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 …]
DTensorInitializer.h31 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,
DTensorForcedEval.h92 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;
DTensorInflation.h78 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;
DTensorGenerator.h81 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);
DTensorPatch.h81 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;
DTensorScan.h84 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;
DTensorRef.h43 …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…
DTensorBroadcasting.h96 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/
Dcxx11_tensor_chipping.cpp348 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()

123