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/
DTensorEvaluator.h28 struct TensorEvaluator struct
66 EIGEN_STRONG_INLINE TensorEvaluator(const Derived& m, const Device& device) in TensorEvaluator() argument
227 struct TensorEvaluator<const Derived, Device>
265 EIGEN_STRONG_INLINE TensorEvaluator(const Derived& m, const Device& device)
356 struct TensorEvaluator<const TensorCwiseNullaryOp<NullaryOp, ArgType>, Device>
360 TensorEvaluator(const XprType& op, const Device& device)
369 typedef typename TensorEvaluator<ArgType, Device>::Dimensions Dimensions;
382 Layout = TensorEvaluator<ArgType, Device>::Layout,
433 TensorEvaluator<ArgType, Device> m_argImpl;
442 struct TensorEvaluator<const TensorCwiseUnaryOp<UnaryOp, ArgType>, Device>
[all …]
DTensorConversion.h53 template <typename TensorEvaluator, typename SrcPacket, typename TgtPacket, int SrcCoeffRatio, int …
56 template <typename TensorEvaluator, typename SrcPacket, typename TgtPacket>
57 struct PacketConverter<TensorEvaluator, SrcPacket, TgtPacket, 1, 1> {
59 PacketConverter(const TensorEvaluator& impl)
68 const TensorEvaluator& m_impl;
72 template <typename TensorEvaluator, typename SrcPacket, typename TgtPacket>
73 struct PacketConverter<TensorEvaluator, SrcPacket, TgtPacket, 2, 1> {
75 PacketConverter(const TensorEvaluator& impl)
89 const TensorEvaluator& m_impl;
92 template <typename TensorEvaluator, typename SrcPacket, typename TgtPacket>
[all …]
DTensorAssign.h92 struct TensorEvaluator<const TensorAssignOp<LeftArgType, RightArgType>, Device>
99 typedef typename TensorEvaluator<RightArgType, Device>::Dimensions Dimensions;
107 IsAligned = int(TensorEvaluator<LeftArgType, Device>::IsAligned) &
108 int(TensorEvaluator<RightArgType, Device>::IsAligned),
109 PacketAccess = int(TensorEvaluator<LeftArgType, Device>::PacketAccess) &
110 int(TensorEvaluator<RightArgType, Device>::PacketAccess),
111 BlockAccess = int(TensorEvaluator<LeftArgType, Device>::BlockAccess) &
112 int(TensorEvaluator<RightArgType, Device>::BlockAccess),
113 PreferBlockAccess = int(TensorEvaluator<LeftArgType, Device>::PreferBlockAccess) |
114 int(TensorEvaluator<RightArgType, Device>::PreferBlockAccess),
[all …]
DTensorLayoutSwap.h95 struct TensorEvaluator<const TensorLayoutSwapOp<ArgType>, Device>
99 …static const int NumDims = internal::array_size<typename TensorEvaluator<ArgType, Device>::Dimensi…
103 IsAligned = TensorEvaluator<ArgType, Device>::IsAligned,
104 PacketAccess = TensorEvaluator<ArgType, Device>::PacketAccess,
106 PreferBlockAccess = TensorEvaluator<ArgType, Device>::PreferBlockAccess,
107 …Layout = (static_cast<int>(TensorEvaluator<ArgType, Device>::Layout) == static_cast<int>(ColMajor)…
109 RawAccess = TensorEvaluator<ArgType, Device>::RawAccess
116 EIGEN_STRONG_INLINE TensorEvaluator(const XprType& op, const Device& device)
165 const TensorEvaluator<ArgType, Device>& impl() const { return m_impl; }
168 TensorEvaluator<ArgType, Device> m_impl;
[all …]
DTensorConcatenation.h95 struct TensorEvaluator<const TensorConcatenationOp<Axis, LeftArgType, RightArgType>, Device>
99 …static const int NumDims = internal::array_size<typename TensorEvaluator<LeftArgType, Device>::Dim…
100 …static const int RightNumDims = internal::array_size<typename TensorEvaluator<RightArgType, Device…
109 PacketAccess = TensorEvaluator<LeftArgType, Device>::PacketAccess &&
110 TensorEvaluator<RightArgType, Device>::PacketAccess,
112 PreferBlockAccess = TensorEvaluator<LeftArgType, Device>::PreferBlockAccess ||
113 TensorEvaluator<RightArgType, Device>::PreferBlockAccess,
114 Layout = TensorEvaluator<LeftArgType, Device>::Layout,
122 EIGEN_STRONG_INLINE TensorEvaluator(const XprType& op, const Device& device)
125 …GEN_STATIC_ASSERT((static_cast<int>(TensorEvaluator<LeftArgType, Device>::Layout) == static_cast<i…
[all …]
DTensorMorphing.h84 struct TensorEvaluator<const TensorReshapingOp<NewDimensions, ArgType>, Device>
98 …static const int NumInputDims = internal::array_size<typename TensorEvaluator<ArgType, Device>::D…
120 IsAligned = TensorEvaluator<ArgType, Device>::IsAligned,
121 PacketAccess = TensorEvaluator<ArgType, Device>::PacketAccess,
125 BlockAccess = TensorEvaluator<ArgType, Device>::RawAccess &&
128 Layout = TensorEvaluator<ArgType, Device>::Layout,
130 RawAccess = TensorEvaluator<ArgType, Device>::RawAccess
145 EIGEN_STRONG_INLINE TensorEvaluator(const XprType& op, const Device& device)
224 EIGEN_DEVICE_FUNC const TensorEvaluator<ArgType, Device>& impl() const { return m_impl; }
233 TensorEvaluator<ArgType, Device> m_impl;
[all …]
DTensorArgMax.h76 struct TensorEvaluator<const TensorIndexTupleOp<ArgType>, Device>
83 typedef typename TensorEvaluator<ArgType, Device>::Dimensions Dimensions;
92 PreferBlockAccess = TensorEvaluator<ArgType, Device>::PreferBlockAccess,
93 Layout = TensorEvaluator<ArgType, Device>::Layout,
102 EIGEN_STRONG_INLINE TensorEvaluator(const XprType& op, const Device& device)
136 TensorEvaluator<ArgType, Device> m_impl;
214 struct TensorEvaluator<const TensorTupleReducerOp<ReduceOp, Dims, ArgType>, Device>
221 …typedef typename TensorEvaluator<const TensorReductionOp<ReduceOp, Dims, const TensorIndexTupleOp<…
222 …typedef typename TensorEvaluator<const TensorIndexTupleOp<ArgType> , Device>::Dimensions InputDime…
233 PreferBlockAccess = TensorEvaluator<ArgType, Device>::PreferBlockAccess,
[all …]
DTensorStriding.h85 struct TensorEvaluator<const TensorStridingOp<Strides, ArgType>, Device>
89 …static const int NumDims = internal::array_size<typename TensorEvaluator<ArgType, Device>::Dimensi…
100 PacketAccess = TensorEvaluator<ArgType, Device>::PacketAccess,
102 PreferBlockAccess = TensorEvaluator<ArgType, Device>::PreferBlockAccess,
103 Layout = TensorEvaluator<ArgType, Device>::Layout,
112 EIGEN_STRONG_INLINE TensorEvaluator(const XprType& op, const Device& device)
120 const typename TensorEvaluator<ArgType, Device>::Dimensions& input_dims = m_impl.dimensions();
257 TensorEvaluator<ArgType, Device> m_impl;
262 struct TensorEvaluator<TensorStridingOp<Strides, ArgType>, Device>
263 : public TensorEvaluator<const TensorStridingOp<Strides, ArgType>, Device>
[all …]
DTensorEvalTo.h98 struct TensorEvaluator<const TensorEvalToOp<ArgType, MakePointer_>, Device>
102 typedef typename TensorEvaluator<ArgType, Device>::Dimensions Dimensions;
111 IsAligned = TensorEvaluator<ArgType, Device>::IsAligned,
112 PacketAccess = TensorEvaluator<ArgType, Device>::PacketAccess,
115 Layout = TensorEvaluator<ArgType, Device>::Layout,
126 typedef typename TensorEvaluator<const ArgType, Device>::TensorBlock
134 EIGEN_STRONG_INLINE TensorEvaluator(const XprType& op, const Device& device)
138 EIGEN_STRONG_INLINE ~TensorEvaluator() {
164 …ype, PacketReturnType, Aligned>(m_buffer + i, m_impl.template packet<TensorEvaluator<ArgType, Devi…
228 TensorEvaluator<ArgType, Device> m_impl;
DTensorChipping.h115 struct TensorEvaluator<const TensorChippingOp<DimId, ArgType>, Device>
118 …static const int NumInputDims = internal::array_size<typename TensorEvaluator<ArgType, Device>::Di…
133 Layout = TensorEvaluator<ArgType, Device>::Layout,
134 PacketAccess = TensorEvaluator<ArgType, Device>::PacketAccess,
135 BlockAccess = TensorEvaluator<ArgType, Device>::BlockAccess,
145 PreferBlockAccess = TensorEvaluator<ArgType, Device>::PreferBlockAccess ||
159 typedef typename TensorEvaluator<const ArgType, Device>::TensorBlock
167 EIGEN_STRONG_INLINE TensorEvaluator(const XprType& op, const Device& device)
173 const typename TensorEvaluator<ArgType, Device>::Dimensions& input_dims = m_impl.dimensions();
402 TensorEvaluator<ArgType, Device> m_impl;
[all …]
DTensorShuffling.h86 struct TensorEvaluator<const TensorShufflingOp<Shuffle, ArgType>, Device>
88 typedef TensorEvaluator<const TensorShufflingOp<Shuffle, ArgType>, Device> Self;
91 …static const int NumDims = internal::array_size<typename TensorEvaluator<ArgType, Device>::Dimensi…
103 BlockAccess = TensorEvaluator<ArgType, Device>::RawAccess,
105 Layout = TensorEvaluator<ArgType, Device>::Layout,
121 EIGEN_STRONG_INLINE TensorEvaluator(const XprType& op, const Device& device)
125 const typename TensorEvaluator<ArgType, Device>::Dimensions& input_dims = m_impl.dimensions();
230 …return PacketLoader<LoadMode, Self, TensorEvaluator<ArgType, Device>::PacketAccess>::Run(*this, in…
353 TensorEvaluator<ArgType, Device> m_impl;
359 struct TensorEvaluator<TensorShufflingOp<Shuffle, ArgType>, Device>
[all …]
DTensorConvolution.h293 struct TensorEvaluator<const TensorConvolutionOp<Indices, InputArgType, KernelArgType>, Device>
297 …static const int NumDims = internal::array_size<typename TensorEvaluator<InputArgType, Device>::Di…
310 …IsAligned = int(TensorEvaluator<InputArgType, Device>::IsAligned) & int(TensorEvaluator<KernelArgT…
311 …PacketAccess = int(TensorEvaluator<InputArgType, Device>::PacketAccess) & int(TensorEvaluator<Kern…
314 Layout = TensorEvaluator<InputArgType, Device>::Layout,
323 EIGEN_STRONG_INLINE TensorEvaluator(const XprType& op, const Device& device)
326 …EN_STATIC_ASSERT((static_cast<int>(TensorEvaluator<InputArgType, Device>::Layout) == static_cast<i…
328 …const typename TensorEvaluator<InputArgType, Device>::Dimensions& input_dims = m_inputImpl.dimensi…
329 …const typename TensorEvaluator<KernelArgType, Device>::Dimensions& kernel_dims = m_kernelImpl.dime…
550 TensorEvaluator<InputArgType, Device> m_inputImpl;
[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,
DTensorReverse.h87 struct TensorEvaluator<const TensorReverseOp<ReverseDimensions, ArgType>, Device>
102 PacketAccess = TensorEvaluator<ArgType, Device>::PacketAccess,
105 Layout = TensorEvaluator<ArgType, Device>::Layout,
116 typedef typename TensorEvaluator<const ArgType, Device>::TensorBlock
124 EIGEN_STRONG_INLINE TensorEvaluator(const XprType& op, const Device& device)
381 TensorEvaluator<ArgType, Device> m_impl;
409 struct TensorEvaluator<TensorReverseOp<ReverseDimensions, ArgType>, Device>
410 : public TensorEvaluator<const TensorReverseOp<ReverseDimensions, ArgType>,
412 typedef TensorEvaluator<const TensorReverseOp<ReverseDimensions, ArgType>,
421 PacketAccess = TensorEvaluator<ArgType, Device>::PacketAccess,
[all …]
DTensorTrace.h82 struct TensorEvaluator<const TensorTraceOp<Dims, ArgType>, Device>
85 …static const int NumInputDims = internal::array_size<typename TensorEvaluator<ArgType, Device>::Di…
99 PacketAccess = TensorEvaluator<ArgType, Device>::PacketAccess,
101 PreferBlockAccess = TensorEvaluator<ArgType, Device>::PreferBlockAccess,
102 Layout = TensorEvaluator<ArgType, Device>::Layout,
111 EIGEN_STRONG_INLINE TensorEvaluator(const XprType& op, const Device& device)
140 const typename TensorEvaluator<ArgType, Device>::Dimensions& input_dims = m_impl.dimensions();
289 TensorEvaluator<ArgType, Device> m_impl;
DTensorInflation.h79 struct TensorEvaluator<const TensorInflationOp<Strides, ArgType>, Device>
83 …static const int NumDims = internal::array_size<typename TensorEvaluator<ArgType, Device>::Dimensi…
94 PacketAccess = TensorEvaluator<ArgType, Device>::PacketAccess,
97 Layout = TensorEvaluator<ArgType, Device>::Layout,
106 EIGEN_STRONG_INLINE TensorEvaluator(const XprType& op, const Device& device)
120 const typename TensorEvaluator<ArgType, Device>::Dimensions& input_dims = m_impl.dimensions();
240 TensorEvaluator<ArgType, Device> m_impl;
DTensorRef.h43 …luatorReadOnly : public TensorLazyBaseEvaluator<Dimensions, typename TensorEvaluator<Expr, Device>…
46 typedef typename TensorEvaluator<Expr, Device>::Scalar Scalar;
49 typedef TensorEvaluator<Expr, Device> EvalType;
75 TensorEvaluator<Expr, Device> m_impl;
367 struct TensorEvaluator<const TensorRef<Derived>, Device>
391 EIGEN_STRONG_INLINE TensorEvaluator(const TensorRef<Derived>& m, const Device&)
420 struct TensorEvaluator<TensorRef<Derived>, Device> : public TensorEvaluator<const TensorRef<Derived…
428 typedef TensorEvaluator<const TensorRef<Derived>, Device> Base;
442 EIGEN_STRONG_INLINE TensorEvaluator(TensorRef<Derived>& m, const Device& d) : Base(m, d)
DTensorConvolutionSycl.h275 struct TensorEvaluator<const TensorConvolutionOp<Indices, InputArgType, KernelArgType>, Eigen::Sycl…
279 …internal::array_size<typename TensorEvaluator<InputArgType, Eigen::SyclDevice>::Dimensions>::value;
283 typedef typename TensorEvaluator<KernelArgType, Eigen::SyclDevice>::Dimensions KernelDimensions;
294 IsAligned = TensorEvaluator<InputArgType, Eigen::SyclDevice>::IsAligned &
295 TensorEvaluator<KernelArgType, Eigen::SyclDevice>::IsAligned,
299 Layout = TensorEvaluator<InputArgType, Eigen::SyclDevice>::Layout,
308 TensorEvaluator(const XprType &op, const Eigen::SyclDevice &device)
317 … EIGEN_STATIC_ASSERT((static_cast<int>(TensorEvaluator<InputArgType, Eigen::SyclDevice>::Layout) ==
318 … static_cast<int>(TensorEvaluator<KernelArgType, Eigen::SyclDevice>::Layout)),
321 …const typename TensorEvaluator<InputArgType, Eigen::SyclDevice>::Dimensions &input_dims = m_inputI…
[all …]
DTensorContraction.h66 struct traits<TensorEvaluator<const TensorContractionOp<Indices_, LeftArgType_, RightArgType_, Outp…
386 Layout = TensorEvaluator<LeftArgType, Device>::Layout,
404 typedef TensorEvaluator<EvalLeftArgType, Device> LeftEvaluatorType;
405 typedef TensorEvaluator<EvalRightArgType, Device> RightEvaluatorType;
408 internal::array_size<typename TensorEvaluator<EvalLeftArgType, Device>::Dimensions>::value;
410 internal::array_size<typename TensorEvaluator<EvalRightArgType, Device>::Dimensions>::value;
429 EIGEN_STATIC_ASSERT((static_cast<int>(TensorEvaluator<LeftArgType, Device>::Layout) ==
430 static_cast<int>(TensorEvaluator<RightArgType, Device>::Layout)),
738 typedef TensorEvaluator<EvalLeftArgType, Device> LeftEvaluator;
739 typedef TensorEvaluator<EvalRightArgType, Device> RightEvaluator;
[all …]
DTensorPatch.h82 struct TensorEvaluator<const TensorPatchOp<PatchDim, ArgType>, Device>
86 …static const int NumDims = internal::array_size<typename TensorEvaluator<ArgType, Device>::Dimensi…
98 PacketAccess = TensorEvaluator<ArgType, Device>::PacketAccess,
100 PreferBlockAccess = TensorEvaluator<ArgType, Device>::PreferBlockAccess,
101 Layout = TensorEvaluator<ArgType, Device>::Layout,
110 EIGEN_STRONG_INLINE TensorEvaluator(const XprType& op, const Device& device)
114 const typename TensorEvaluator<ArgType, Device>::Dimensions& input_dims = m_impl.dimensions();
285 TensorEvaluator<ArgType, Device> m_impl;
DTensorForwardDeclarations.h98 template<typename Derived, typename Device> struct TensorEvaluator;
148 static const bool value = TensorEvaluator<Expression, Device>::PacketAccess;
153 static const bool value = TensorEvaluator<Expression, GpuDevice>::PacketAccess &&
154 TensorEvaluator<Expression, GpuDevice>::IsAligned;
169 TensorEvaluator<Expression, Device>::BlockAccess &&
170 TensorEvaluator<Expression, Device>::PreferBlockAccess;
DTensorExecutor.h98 TensorEvaluator<Expression, Device> evaluator(expr, device);
130 TensorEvaluator<Expression, DefaultDevice> evaluator(expr, device);
134 const int PacketSize = unpacket_traits<typename TensorEvaluator<
170 typedef TensorEvaluator<Expression, DefaultDevice> Evaluator;
331 typedef TensorEvaluator<Expression, ThreadPoolDevice> Evaluator;
358 typedef TensorEvaluator<Expression, ThreadPoolDevice> Evaluator;
409 typedef TensorEvaluator<Expression, ThreadPoolDevice> Evaluator;
466 typedef TensorEvaluator<Expression, ThreadPoolDevice> Evaluator;
599 TensorEvaluator<Expression, GpuDevice> evaluator(expr, device);
611 (EigenMetaKernel<TensorEvaluator<Expression, GpuDevice>, StorageIndex>),
[all …]
DTensorForcedEval.h104 struct TensorEvaluator<const TensorForcedEvalOp<ArgType_>, Device>
109 typedef typename TensorEvaluator<ArgType, Device>::Dimensions Dimensions;
123 Layout = TensorEvaluator<ArgType, Device>::Layout,
138 TensorEvaluator(const XprType& op, const Device& device)
228 TensorEvaluator<ArgType, Device> m_impl;
DTensorImagePatch.h214 struct TensorEvaluator<const TensorImagePatchOp<Rows, Cols, ArgType>, Device>
218 …static const int NumInputDims = internal::array_size<typename TensorEvaluator<ArgType, Device>::Di…
222 typedef TensorEvaluator<const TensorImagePatchOp<Rows, Cols, ArgType>,
224 typedef TensorEvaluator<ArgType, Device> Impl;
233 PacketAccess = TensorEvaluator<ArgType, Device>::PacketAccess,
236 Layout = TensorEvaluator<ArgType, Device>::Layout,
245 EIGEN_STRONG_INLINE TensorEvaluator( const XprType& op, const Device& device)
252 const typename TensorEvaluator<ArgType, Device>::Dimensions& input_dims = m_impl.dimensions();
508 …EIGEN_DEVICE_FUNC EIGEN_STRONG_INLINE const TensorEvaluator<ArgType, Device>& impl() const { retur…
597 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