| /external/eigen/unsupported/Eigen/CXX11/src/Tensor/ |
| 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 internal::pstoret<CoeffReturnType, PacketReturnType, Aligned>(m_buffer + i, m_impl.template packet<TensorEvaluator<ArgType, Device>::IsAligned ? Aligned : Unaligned>(i));
166 const TensorEvaluator<ArgType, Device>& impl() const { return m_impl; }
171 TensorEvaluator<ArgType, Device> m_impl
[all...] |
| TensorArgMax.h | 76 struct TensorEvaluator<const TensorIndexTupleOp<ArgType>, Device>
83 typedef typename TensorEvaluator<ArgType, Device>::Dimensions Dimensions;
87 IsAligned = /*TensorEvaluator<ArgType, Device>::IsAligned*/ false,
88 PacketAccess = /*TensorEvaluator<ArgType, Device>::PacketAccess*/ false,
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<ArgType> >, Device>::Dimensions Dimensions;
209 typedef typename TensorEvaluator<const TensorIndexTupleOp<ArgType> , Device>::Dimensions InputDimensions
[all...] |
| 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>::IsAligned,
101 PacketAccess = TensorEvaluator<LeftArgType, Device>::PacketAccess & TensorEvaluator<RightArgType, Device>::PacketAccess,
102 Layout = TensorEvaluator<LeftArgType, Device>::Layout,
103 RawAccess = TensorEvaluator<LeftArgType, Device>::RawAccess
106 EIGEN_DEVICE_FUNC TensorEvaluator(const XprType& op, const Device& device) :
110 EIGEN_STATIC_ASSERT((static_cast<int>(TensorEvaluator<LeftArgType, Device>::Layout) == static_cast<int>(TensorEvaluator<RightArgType, Device>::Layout)), (…)
[all...] |
| TensorBroadcasting.h | 96 struct TensorEvaluator<const TensorBroadcastingOp<Broadcast, ArgType>, Device>
100 static const int NumDims = internal::array_size<typename TensorEvaluator<ArgType, Device>::Dimensions>::value;
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;
|
| TensorConversion.h | 52 template <typename TensorEvaluator, typename SrcPacket, typename TgtPacket, int SrcCoeffRatio, int TgtCoeffRatio>
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...] |
| TensorCustomOp.h | 80 struct TensorEvaluator<const TensorCustomUnaryOp<CustomUnaryFunc, XprType>, Device>
95 Layout = TensorEvaluator<XprType, Device>::Layout,
100 EIGEN_DEVICE_FUNC EIGEN_STRONG_INLINE TensorEvaluator(const ArgType& op, const Device& device)
236 struct TensorEvaluator<const TensorCustomBinaryOp<CustomBinaryFunc, LhsXprType, RhsXprType>, Device>
251 Layout = TensorEvaluator<LhsXprType, Device>::Layout,
256 EIGEN_DEVICE_FUNC EIGEN_STRONG_INLINE TensorEvaluator(const XprType& op, const Device& device)
|
| TensorEvaluator.h | 15 /** \class TensorEvaluator
28 struct TensorEvaluator
48 EIGEN_DEVICE_FUNC EIGEN_STRONG_INLINE TensorEvaluator(const Derived& m, const Device& device)
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;
[all...] |
| 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;
|
| 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);
|
| TensorInflation.h | 78 struct TensorEvaluator<const TensorInflationOp<Strides, ArgType>, Device>
82 static const int NumDims = internal::array_size<typename TensorEvaluator<ArgType, Device>::Dimensions>::value;
90 IsAligned = /*TensorEvaluator<ArgType, Device>::IsAligned*/ false,
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;
|
| TensorLayoutSwap.h | 111 struct TensorEvaluator<const TensorLayoutSwapOp<ArgType>, Device>
115 static const int NumDims = internal::array_size<typename TensorEvaluator<ArgType, Device>::Dimensions>::value;
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)) ? RowMajor : 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...] |
| TensorPatch.h | 81 struct TensorEvaluator<const TensorPatchOp<PatchDim, ArgType>, Device>
85 static const int NumDims = internal::array_size<typename TensorEvaluator<ArgType, Device>::Dimensions>::value + 1;
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;
|
| TensorRef.h | 43 class TensorLazyEvaluatorReadOnly : public TensorLazyBaseEvaluator<Dimensions, typename TensorEvaluator<Expr, Device>::Scalar> {
45 // typedef typename TensorEvaluator<Expr, Device>::Dimensions Dimensions;
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>, Device>
409 typedef TensorEvaluator<const TensorRef<Derived>, Device> Base;
417 EIGEN_DEVICE_FUNC EIGEN_STRONG_INLINE TensorEvaluator(TensorRef<Derived>& m, const Device& d) : Base(m, d
[all...] |
| 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...] |
| TensorShuffling.h | 100 struct TensorEvaluator<const TensorShufflingOp<Shuffle, ArgType>, Device>
104 static const int NumDims = internal::array_size<typename TensorEvaluator<ArgType, Device>::Dimensions>::value;
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>::Dimensions>::value
[all...] |
| TensorStriding.h | 100 struct TensorEvaluator<const TensorStridingOp<Strides, ArgType>, Device>
104 static const int NumDims = internal::array_size<typename TensorEvaluator<ArgType, Device>::Dimensions>::value;
112 IsAligned = /*TensorEvaluator<ArgType, Device>::IsAligned*/false,
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
[all...] |
| TensorChipping.h | 129 struct TensorEvaluator<const TensorChippingOp<DimId, ArgType>, Device>
132 static const int NumInputDims = internal::array_size<typename TensorEvaluator<ArgType, Device>::Dimensions>::value;
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...] |
| TensorConcatenation.h | 110 struct TensorEvaluator<const TensorConcatenationOp<Axis, LeftArgType, RightArgType>, Device>
114 static const int NumDims = internal::array_size<typename TensorEvaluator<LeftArgType, Device>::Dimensions>::value;
115 static const int RightNumDims = internal::array_size<typename TensorEvaluator<RightArgType, Device>::Dimensions>::value;
122 PacketAccess = TensorEvaluator<LeftArgType, Device>::PacketAccess & TensorEvaluator<RightArgType, Device>::PacketAccess,
123 Layout = TensorEvaluator<LeftArgType, Device>::Layout,
127 EIGEN_DEVICE_FUNC EIGEN_STRONG_INLINE TensorEvaluator(const XprType& op, const Device& device)
130 EIGEN_STATIC_ASSERT((static_cast<int>(TensorEvaluator<LeftArgType, Device>::Layout) == static_cast<int>(TensorEvaluator<RightArgType, Device>::Layout) || NumDims == 1), YOU_MADE_A_PROGRAMMING_MISTAKE);
285 TensorEvaluator<LeftArgType, Device> m_leftImpl
[all...] |
| TensorPadding.h | 84 struct TensorEvaluator<const TensorPaddingOp<PaddingDimensions, ArgType>, Device>
97 PacketAccess = TensorEvaluator<ArgType, Device>::PacketAccess,
98 Layout = TensorEvaluator<ArgType, Device>::Layout,
103 EIGEN_DEVICE_FUNC EIGEN_STRONG_INLINE TensorEvaluator(const XprType& op, const Device& device)
116 const typename TensorEvaluator<ArgType, Device>::Dimensions& input_dims = m_impl.dimensions();
386 TensorEvaluator<ArgType, Device> m_impl;
|
| TensorScan.h | 84 struct TensorEvaluator<const TensorScanOp<Op, ArgType>, Device> {
88 static const int NumDims = internal::array_size<typename TensorEvaluator<ArgType, Device>::Dimensions>::value;
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;
|
| TensorFFT.h | 118 struct TensorEvaluator<const TensorFFTOp<FFT, ArgType, FFTResultType, FFTDir>, Device> {
121 static const int NumDims = internal::array_size<typename TensorEvaluator<ArgType, Device>::Dimensions>::value;
126 typedef typename TensorEvaluator<ArgType, Device>::Dimensions InputDimensions;
138 Layout = TensorEvaluator<ArgType, Device>::Layout,
143 EIGEN_DEVICE_FUNC EIGEN_STRONG_INLINE TensorEvaluator(const XprType& op, const Device& device) : m_fft(op.fft()), m_impl(op.expression(), device), m_data(NULL), m_device(device) {
144 const typename TensorEvaluator<ArgType, Device>::Dimensions& input_dims = m_impl.dimensions();
568 TensorEvaluator<ArgType, Device> m_impl;
|
| 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...] |
| TensorReduction.h | 382 struct TensorEvaluator<const TensorReductionOp<Op, Dims, ArgType, MakePointer_>, Device>
387 typedef typename TensorEvaluator<ArgType, Device>::Dimensions InputDimensions;
393 typedef TensorEvaluator<const TensorReductionOp<Op, Dims, ArgType, MakePointer_>, Device> Self;
394 static const bool InputPacketAccess = TensorEvaluator<ArgType, Device>::PacketAccess;
402 Layout = TensorEvaluator<ArgType, Device>::Layout,
411 EIGEN_DEVICE_FUNC EIGEN_STRONG_INLINE TensorEvaluator(const XprType& op, const Device& device)
428 const typename TensorEvaluator<ArgType, Device>::Dimensions& input_dims = m_impl.dimensions();
667 const TensorEvaluator<ArgType, Device>& impl() const { return m_impl; }
[all...] |
| TensorContraction.h | 62 struct traits<TensorEvaluator<const TensorContractionOp<Indices_, LeftArgType_, RightArgType_>, Device_> > {
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...] |
| /external/tensorflow/tensorflow/core/kernels/ |
| mirror_pad_op.h | 91 struct TensorEvaluator<const TensorMirrorPadOp<PaddingDimensions, ArgType>,
104 PacketAccess = TensorEvaluator<ArgType, Device>::PacketAccess,
106 Layout = TensorEvaluator<ArgType, Device>::Layout,
111 EIGEN_DEVICE_FUNC EIGEN_STRONG_INLINE TensorEvaluator(const XprType& op,
172 ReadInputHelper<TensorEvaluator<ArgType, Device>::CoordAccess> helper;
322 TensorEvaluator<ArgType, Device> impl_;
|
