| /external/eigen/unsupported/Eigen/CXX11/src/Tensor/ |
| TensorCustomOp.h | 57 typedef typename XprType::CoeffReturnType CoeffReturnType;
87 typedef typename internal::remove_const<typename XprType::CoeffReturnType>::type CoeffReturnType;
88 typedef typename PacketType<CoeffReturnType, Device>::type PacketReturnType;
108 EIGEN_DEVICE_FUNC EIGEN_STRONG_INLINE bool evalSubExprsIfNeeded(CoeffReturnType* data) {
113 m_result = static_cast<CoeffReturnType*>(
127 EIGEN_DEVICE_FUNC EIGEN_STRONG_INLINE CoeffReturnType coeff(Index index) const {
138 return TensorOpCost(sizeof(CoeffReturnType), 0, 0, vectorized, PacketSize);
141 EIGEN_DEVICE_FUNC CoeffReturnType* data() const { return m_result;
[all...] |
| TensorForcedEval.h | 74 typedef typename internal::remove_const<typename XprType::CoeffReturnType>::type CoeffReturnType;
98 typedef typename XprType::CoeffReturnType CoeffReturnType;
99 typedef typename PacketType<CoeffReturnType, Device>::type PacketReturnType;
116 EIGEN_DEVICE_FUNC EIGEN_STRONG_INLINE bool evalSubExprsIfNeeded(CoeffReturnType*) {
118 m_buffer = (CoeffReturnType*)m_device.allocate(numValues * sizeof(CoeffReturnType));
120 if (NumTraits<CoeffReturnType>::RequireInitialization) {
122 new(m_buffer+i) CoeffReturnType();
[all...] |
| TensorScan.h | 54 typedef typename XprType::CoeffReturnType CoeffReturnType;
91 typedef typename XprType::CoeffReturnType CoeffReturnType;
92 typedef typename PacketType<CoeffReturnType, Device>::type PacketReturnType;
168 m_output = static_cast<CoeffReturnType*>(m_device.allocate(total_size * sizeof(Scalar)));
178 EIGEN_DEVICE_FUNC EIGEN_STRONG_INLINE CoeffReturnType* data() const
183 EIGEN_DEVICE_FUNC EIGEN_STRONG_INLINE CoeffReturnType coeff(Index index) const
189 return TensorOpCost(sizeof(CoeffReturnType), 0, 0);
207 CoeffReturnType* m_output
[all...] |
| TensorLayoutSwap.h | 73 typedef typename internal::remove_const<typename XprType::CoeffReturnType>::type CoeffReturnType;
135 typedef typename XprType::CoeffReturnType CoeffReturnType;
136 typedef typename PacketType<CoeffReturnType, Device>::type PacketReturnType;
140 EIGEN_DEVICE_FUNC EIGEN_STRONG_INLINE bool evalSubExprsIfNeeded(CoeffReturnType* data) {
147 EIGEN_DEVICE_FUNC EIGEN_STRONG_INLINE CoeffReturnType coeff(Index index) const
193 typedef typename XprType::CoeffReturnType CoeffReturnType;
194 typedef typename PacketType<CoeffReturnType, Device>::type PacketReturnType
[all...] |
| TensorEvalTo.h | 70 typedef typename internal::remove_const<typename XprType::CoeffReturnType>::type CoeffReturnType;
71 typedef typename MakePointer_<CoeffReturnType>::Type PointerType;
99 typedef typename internal::remove_const<typename XprType::CoeffReturnType>::type CoeffReturnType;
100 typedef typename PacketType<CoeffReturnType, Device>::type PacketReturnType;
124 typedef typename internal::traits<const TensorEvalToOp<ArgType, MakePointer_> >::template MakePointer<CoeffReturnType>::Type DevicePointer;
137 internal::pstoret<CoeffReturnType, PacketReturnType, Aligned>(m_buffer + i, m_impl.template packet<TensorEvaluator<ArgType, Device>::IsAligned ? Aligned : Unaligned>(i));
144 EIGEN_DEVICE_FUNC EIGEN_STRONG_INLINE CoeffReturnType coeff(Index index) const
159 TensorOpCost(0, sizeof(CoeffReturnType), 0, vectorized, PacketSize)
[all...] |
| TensorReductionSycl.h | 28 template<typename CoeffReturnType, typename KernelName> struct syclGenericBufferReducer{
41 cl::sycl::accessor<CoeffReturnType, 1, cl::sycl::access::mode::read_write,
105 typedef typename Self::CoeffReturnType CoeffReturnType;
108 static void run(const Self& self, Op& reducer, const Eigen::SyclDevice& dev, CoeffReturnType* output) {
137 auto out_buffer =dev.template get_sycl_buffer<typename Eigen::internal::remove_all<CoeffReturnType>::type>(self.dimensions().TotalSize(), output);
141 auto temp_global_buffer =cl::sycl::buffer<CoeffReturnType, 1>(cl::sycl::range<1>(GRange));
166 tmp_global_accessor.get_pointer()[globalid]=static_cast<CoeffReturnType>(0);
176 syclGenericBufferReducer<CoeffReturnType,HostExpr>::run(out_buffer, temp_global_buffer,dev, GRange, outTileSize);
184 typedef typename Self::CoeffReturnType CoeffReturnType
[all...] |
| TensorShuffling.h | 58 typedef typename XprType::CoeffReturnType CoeffReturnType;
107 typedef typename XprType::CoeffReturnType CoeffReturnType;
108 typedef typename PacketType<CoeffReturnType, Device>::type PacketReturnType;
161 EIGEN_DEVICE_FUNC EIGEN_STRONG_INLINE CoeffReturnType coeff(Index index) const
172 EIGEN_ALIGN_MAX typename internal::remove_const<CoeffReturnType>::type values[PacketSize];
229 typedef typename XprType::CoeffReturnType CoeffReturnType;
230 typedef typename PacketType<CoeffReturnType, Device>::type PacketReturnType
[all...] |
| TensorEvaluator.h | 32 typedef typename Derived::Scalar CoeffReturnType;
33 typedef typename PacketType<CoeffReturnType, Device>::type PacketReturnType;
56 EIGEN_DEVICE_FUNC EIGEN_STRONG_INLINE bool evalSubExprsIfNeeded(CoeffReturnType* dest) {
66 EIGEN_DEVICE_FUNC EIGEN_STRONG_INLINE CoeffReturnType coeff(Index index) const {
88 EIGEN_DEVICE_FUNC EIGEN_STRONG_INLINE CoeffReturnType coeff(const array<DenseIndex, NumCoords>& coords) const {
107 return TensorOpCost(sizeof(CoeffReturnType), 0, 0, vectorized,
152 typedef typename Derived::Scalar CoeffReturnType;
153 typedef typename PacketType<CoeffReturnType, Device>::type PacketReturnType;
177 EIGEN_DEVICE_FUNC EIGEN_STRONG_INLINE bool evalSubExprsIfNeeded(CoeffReturnType* data) {
187 EIGEN_DEVICE_FUNC EIGEN_STRONG_INLINE CoeffReturnType coeff(Index index) const
[all...] |
| TensorReverse.h | 58 typedef typename XprType::CoeffReturnType CoeffReturnType;
108 typedef typename XprType::CoeffReturnType CoeffReturnType;
109 typedef typename PacketType<CoeffReturnType, Device>::type PacketReturnType;
189 EIGEN_DEVICE_FUNC EIGEN_STRONG_INLINE CoeffReturnType coeff(
203 EIGEN_ALIGN_MAX typename internal::remove_const<CoeffReturnType>::type
259 typedef typename XprType::CoeffReturnType CoeffReturnType;
260 typedef typename PacketType<CoeffReturnType, Device>::type PacketReturnType
[all...] |
| TensorAssign.h | 65 typedef typename LhsXprType::CoeffReturnType CoeffReturnType;
94 typedef typename XprType::CoeffReturnType CoeffReturnType;
95 typedef typename PacketType<CoeffReturnType, Device>::type PacketReturnType;
143 EIGEN_DEVICE_FUNC CoeffReturnType coeff(Index index) const
161 numext::maxi(0.0, left.bytes_loaded() - sizeof(CoeffReturnType)),
163 TensorOpCost(0, sizeof(CoeffReturnType), 0, vectorized, PacketSize);
171 EIGEN_DEVICE_FUNC CoeffReturnType* data() const { return m_leftImpl.data(); }
|
| TensorArgMax.h | 61 typedef Tuple<Index, typename XprType::CoeffReturnType> CoeffReturnType;
81 typedef typename XprType::CoeffReturnType CoeffReturnType;
110 EIGEN_DEVICE_FUNC EIGEN_STRONG_INLINE CoeffReturnType coeff(Index index) const
112 return CoeffReturnType(index, m_impl.coeff(index));
171 typedef Index CoeffReturnType;
206 typedef typename XprType::CoeffReturnType CoeffReturnType;
207 typedef typename TensorIndexTupleOp<ArgType>::CoeffReturnType TupleType
[all...] |
| TensorReduction.h | 131 static EIGEN_DEVICE_FUNC EIGEN_STRONG_INLINE void reduce(const Self& self, typename Self::Index firstIndex, Op& reducer, typename Self::CoeffReturnType* accum) {
141 static EIGEN_DEVICE_FUNC EIGEN_STRONG_INLINE void reduce(const Self& self, typename Self::Index firstIndex, Op& reducer, typename Self::CoeffReturnType* accum) {
150 static EIGEN_DEVICE_FUNC EIGEN_STRONG_INLINE void reduce(const Self& self, typename Self::Index index, Op& reducer, typename Self::CoeffReturnType* accum) {
157 static EIGEN_DEVICE_FUNC EIGEN_STRONG_INLINE typename Self::CoeffReturnType reduce(const Self& self, typename Self::Index firstIndex, typename Self::Index numValuesToReduce, Op& reducer) {
158 typename Self::CoeffReturnType accum = reducer.initialize();
168 static EIGEN_DEVICE_FUNC EIGEN_STRONG_INLINE typename Self::CoeffReturnType reduce(const Self& self, typename Self::Index firstIndex, typename Self::Index numValuesToReduce, Op& reducer) {
175 typename Self::CoeffReturnType accum = reducer.initialize();
222 static EIGEN_DEVICE_FUNC void run(const Self& self, Op& reducer, const Device&, typename Self::CoeffReturnType* output) {
236 typename Self::CoeffReturnType* output) {
251 typename Self::CoeffReturnType* output)
[all...] |
| TensorChipping.h | 83 typedef typename XprType::CoeffReturnType CoeffReturnType;
137 typedef typename XprType::CoeffReturnType CoeffReturnType;
138 typedef typename PacketType<CoeffReturnType, Device>::type PacketReturnType;
197 EIGEN_DEVICE_FUNC EIGEN_STRONG_INLINE CoeffReturnType coeff(Index index) const
213 EIGEN_ALIGN_MAX typename internal::remove_const<CoeffReturnType>::type values[PacketSize];
233 EIGEN_ALIGN_MAX typename internal::remove_const<CoeffReturnType>::type values[PacketSize];
266 EIGEN_DEVICE_FUNC EIGEN_STRONG_INLINE CoeffReturnType* data() const {
267 CoeffReturnType* result = const_cast<CoeffReturnType*>(m_impl.data())
[all...] |
| TensorGenerator.h | 58 typedef typename XprType::CoeffReturnType CoeffReturnType;
88 typedef typename XprType::CoeffReturnType CoeffReturnType;
89 typedef typename PacketType<CoeffReturnType, Device>::type PacketReturnType;
126 EIGEN_DEVICE_FUNC EIGEN_STRONG_INLINE CoeffReturnType coeff(Index index) const
140 EIGEN_ALIGN_MAX typename internal::remove_const<CoeffReturnType>::type values[packetSize];
|
| TensorMorphing.h | 57 typedef typename internal::remove_const<typename XprType::CoeffReturnType>::type CoeffReturnType;
122 typedef typename XprType::CoeffReturnType CoeffReturnType;
123 typedef typename PacketType<CoeffReturnType, Device>::type PacketReturnType;
127 EIGEN_DEVICE_FUNC EIGEN_STRONG_INLINE bool evalSubExprsIfNeeded(CoeffReturnType* data) {
134 EIGEN_DEVICE_FUNC EIGEN_STRONG_INLINE CoeffReturnType coeff(Index index) const
183 typedef typename XprType::CoeffReturnType CoeffReturnType;
184 typedef typename PacketType<CoeffReturnType, Device>::type PacketReturnType
[all...] |
| TensorBase.h | 33 typedef typename internal::remove_const<Scalar>::type CoeffReturnType;
502 typedef TensorScanOp<internal::SumReducer<CoeffReturnType>, const Derived> TensorScanSumOp;
509 typedef TensorScanOp<internal::ProdReducer<CoeffReturnType>, const Derived> TensorScanProdOp;
525 const TensorReductionOp<internal::SumReducer<CoeffReturnType>, const Dims, const Derived>
527 return TensorReductionOp<internal::SumReducer<CoeffReturnType>, const Dims, const Derived>(derived(), dims, internal::SumReducer<CoeffReturnType>());
530 const TensorReductionOp<internal::SumReducer<CoeffReturnType>, const DimensionList<Index, NumDimensions>, const Derived>
533 return TensorReductionOp<internal::SumReducer<CoeffReturnType>, const DimensionList<Index, NumDimensions>, const Derived>(derived(), in_dims, internal::SumReducer<CoeffReturnType>());
537 const TensorReductionOp<internal::MeanReducer<CoeffReturnType>, const Dims, const Derived
[all...] |
| TensorInflation.h | 56 typedef typename XprType::CoeffReturnType CoeffReturnType;
85 typedef typename XprType::CoeffReturnType CoeffReturnType;
86 typedef typename PacketType<CoeffReturnType, Device>::type PacketReturnType;
177 EIGEN_DEVICE_FUNC EIGEN_STRONG_INLINE CoeffReturnType coeff(Index index) const
195 EIGEN_ALIGN_MAX typename internal::remove_const<CoeffReturnType>::type values[PacketSize];
212 TensorOpCost(sizeof(CoeffReturnType) * input_size / output_size, 0,
|
| TensorBroadcasting.h | 73 typedef typename XprType::CoeffReturnType CoeffReturnType;
104 typedef typename XprType::CoeffReturnType CoeffReturnType;
105 typedef typename PacketType<CoeffReturnType, Device>::type PacketReturnType;
157 EIGEN_DEVICE_FUNC EIGEN_ALWAYS_INLINE CoeffReturnType coeff(Index index) const
171 EIGEN_DEVICE_FUNC EIGEN_STRONG_INLINE CoeffReturnType coeffColMajor(Index index) const
201 EIGEN_DEVICE_FUNC EIGEN_STRONG_INLINE CoeffReturnType coeffRowMajor(Index index) const
289 EIGEN_ALIGN_MAX typename internal::remove_const<CoeffReturnType>::type values[PacketSize];
341 EIGEN_ALIGN_MAX typename internal::remove_const<CoeffReturnType>::type values[PacketSize]
[all...] |
| TensorStriding.h | 58 typedef typename XprType::CoeffReturnType CoeffReturnType;
107 typedef typename XprType::CoeffReturnType CoeffReturnType;
108 typedef typename PacketType<CoeffReturnType, Device>::type PacketReturnType;
159 EIGEN_DEVICE_FUNC EIGEN_STRONG_INLINE CoeffReturnType coeff(Index index) const
200 EIGEN_ALIGN_MAX typename internal::remove_const<CoeffReturnType>::type values[PacketSize];
280 typedef typename XprType::CoeffReturnType CoeffReturnType;
281 typedef typename PacketType<CoeffReturnType, Device>::type PacketReturnType
[all...] |
| TensorConcatenation.h | 65 typedef typename internal::promote_storage_type<typename LhsXprType::CoeffReturnType,
66 typename RhsXprType::CoeffReturnType>::ret CoeffReturnType;
118 typedef typename XprType::CoeffReturnType CoeffReturnType;
119 typedef typename PacketType<CoeffReturnType, Device>::type PacketReturnType;
195 EIGEN_DEVICE_FUNC EIGEN_STRONG_INLINE CoeffReturnType coeff(Index index) const
255 EIGEN_ALIGN_MAX CoeffReturnType values[packetSize];
313 typedef typename XprType::CoeffReturnType CoeffReturnType;
[all...] |
| TensorReductionCuda.h | 126 typename Self::CoeffReturnType* output, unsigned int* semaphore) {
160 typename Self::CoeffReturnType accum = reducer.initialize();
165 typename Self::CoeffReturnType val = input.m_impl.coeff(index);
290 typedef typename Self::CoeffReturnType Scalar;
349 (internal::is_same<typename Self::CoeffReturnType, float>::value ||
350 internal::is_same<typename Self::CoeffReturnType, double>::value ||
351 (internal::is_same<typename Self::CoeffReturnType, Eigen::half>::value && reducer_traits<Op, GpuDevice>::PacketAccess));
354 (internal::is_same<typename Self::CoeffReturnType, float>::value ||
355 internal::is_same<typename Self::CoeffReturnType, double>::value);
375 typename Self::CoeffReturnType* output)
[all...] |
| /external/eigen/Eigen/src/Core/ |
| DenseCoeffsBase.h | 41 // Explanation for this CoeffReturnType typedef.
51 >::type CoeffReturnType;
96 EIGEN_STRONG_INLINE CoeffReturnType coeff(Index row, Index col) const
104 EIGEN_STRONG_INLINE CoeffReturnType coeffByOuterInner(Index outer, Index inner) const
115 EIGEN_STRONG_INLINE CoeffReturnType operator()(Index row, Index col) const
138 EIGEN_STRONG_INLINE CoeffReturnType
157 EIGEN_STRONG_INLINE CoeffReturnType
177 EIGEN_STRONG_INLINE CoeffReturnType
187 EIGEN_STRONG_INLINE CoeffReturnType
193 EIGEN_STRONG_INLINE CoeffReturnType
[all...] |
| NestByValue.h | 49 EIGEN_DEVICE_FUNC inline const CoeffReturnType coeff(Index row, Index col) const
59 EIGEN_DEVICE_FUNC inline const CoeffReturnType coeff(Index index) const
|
| CoreEvaluators.h | 132 typedef typename PlainObjectType::CoeffReturnType CoeffReturnType;
161 CoeffReturnType coeff(Index row, Index col) const
170 CoeffReturnType coeff(Index index) const
278 typedef typename XprType::CoeffReturnType CoeffReturnType;
281 CoeffReturnType coeff(Index row, Index col) const
287 CoeffReturnType coeff(Index index) const
478 typedef typename XprType::CoeffReturnType CoeffReturnType;
[all...] |
| ForceAlignedAccess.h | 49 EIGEN_DEVICE_FUNC inline const CoeffReturnType coeff(Index row, Index col) const
59 EIGEN_DEVICE_FUNC inline const CoeffReturnType coeff(Index index) const
|
