Searched refs:Dimensions (Results 1 - 25 of 63) sorted by relevance

123

/external/eigen/unsupported/Eigen/CXX11/src/Tensor/
H A DTensorTraits.h65 template<typename Scalar_, typename Dimensions, int Options_, typename IndexType_>
66 struct traits<TensorFixedSize<Scalar_, Dimensions, Options_, IndexType_> >
71 static const int NumDimensions = array_size<Dimensions>::value;
133 template<typename Scalar_, typename Dimensions, int Options, typename IndexType_>
134 struct eval<TensorFixedSize<Scalar_, Dimensions, Options, IndexType_>, Eigen::Dense>
136 typedef const TensorFixedSize<Scalar_, Dimensions, Options, IndexType_>& type;
139 template<typename Scalar_, typename Dimensions, int Options, typename IndexType_>
140 struct eval<const TensorFixedSize<Scalar_, Dimensions, Options, IndexType_>, Eigen::Dense>
142 typedef const TensorFixedSize<Scalar_, Dimensions, Options, IndexType_>& type;
187 template <typename Scalar_, typename Dimensions, in
[all...]
H A DTensorRef.h17 template <typename Dimensions, typename Scalar>
23 EIGEN_DEVICE_FUNC virtual const Dimensions& dimensions() const = 0;
42 template <typename Dimensions, typename Expr, typename Device>
43 class TensorLazyEvaluatorReadOnly : public TensorLazyBaseEvaluator<Dimensions, typename TensorEvaluator<Expr, Device>::Scalar> {
45 // typedef typename TensorEvaluator<Expr, Device>::Dimensions Dimensions;
56 EIGEN_DEVICE_FUNC virtual const Dimensions& dimensions() const {
73 Dimensions m_dims;
77 template <typename Dimensions, typename Expr, typename Device>
78 class TensorLazyEvaluatorWritable : public TensorLazyEvaluatorReadOnly<Dimensions, Exp
134 typedef typename PlainObjectType::Dimensions Dimensions; typedef in class:Eigen::TensorRef
362 typedef typename Derived::Dimensions Dimensions; typedef in struct:Eigen::TensorEvaluator
407 typedef typename Derived::Dimensions Dimensions; typedef in struct:Eigen::TensorEvaluator
[all...]
H A DTensorIO.h61 typedef typename Evaluator::Dimensions Dimensions; typedef
69 static const int rank = internal::array_size<Dimensions>::value;
H A DTensorSyclExtractFunctors.h37 typedef typename Evaluator::Dimensions Dimensions; typedef in struct:Eigen::TensorSycl::internal::FunctorExtractor
38 const Dimensions m_dimensions;
39 const Dimensions& dimensions() const { return m_dimensions; }
157 typedef typename Eigen::internal::conditional<Evaluator::NumOutputDims==0, DSizes<typename Evaluator::Index, 1>, typename Evaluator::Dimensions >::type Dimensions; typedef in struct:Eigen::TensorSycl::internal::FunctorExtractor
158 const Dimensions m_dimensions;
159 const Dimensions& dimensions() const { return m_dimensions; }
161 : m_dimensions(DimConstr<Dimensions, Evaluator::NumOutputDims>::getDim(expr.dimensions())) {}
H A DTensorStorage.h34 template<typename T, typename Dimensions, int Options_> class TensorStorage;
74 typedef DSizes<IndexType, NumIndices_> Dimensions; typedef in class:Eigen::TensorStorage
114 EIGEN_DEVICE_FUNC EIGEN_STRONG_INLINE const Dimensions& dimensions() const {return m_dimensions;}
141 Dimensions m_dimensions;
H A DTensorEvaluator.h34 typedef typename Derived::Dimensions Dimensions; typedef in struct:Eigen::TensorEvaluator
54 EIGEN_DEVICE_FUNC EIGEN_STRONG_INLINE const Dimensions& dimensions() const { return m_dims; }
118 Dimensions m_dims;
154 typedef typename Derived::Dimensions Dimensions; typedef in struct:Eigen::TensorEvaluator
175 EIGEN_DEVICE_FUNC EIGEN_STRONG_INLINE const Dimensions& dimensions() const { return m_dims; }
217 Dimensions m_dims;
250 typedef typename TensorEvaluator<ArgType, Device>::Dimensions Dimensions; typedef in struct:Eigen::TensorEvaluator
315 typedef typename TensorEvaluator<ArgType, Device>::Dimensions Dimensions; typedef in struct:Eigen::TensorEvaluator
388 typedef typename TensorEvaluator<LeftArgType, Device>::Dimensions Dimensions; typedef in struct:Eigen::TensorEvaluator
483 typedef typename TensorEvaluator<Arg1Type, Device>::Dimensions Dimensions; typedef in struct:Eigen::TensorEvaluator
573 typedef typename TensorEvaluator<IfArgType, Device>::Dimensions Dimensions; typedef in struct:Eigen::TensorEvaluator
[all...]
H A DTensorShuffling.h104 static const int NumDims = internal::array_size<typename TensorEvaluator<ArgType, Device>::Dimensions>::value;
105 typedef DSizes<Index, NumDims> Dimensions; typedef in struct:Eigen::TensorEvaluator
122 const typename TensorEvaluator<ArgType, Device>::Dimensions& input_dims = m_impl.dimensions();
151 EIGEN_DEVICE_FUNC EIGEN_STRONG_INLINE const Dimensions& dimensions() const { return m_dimensions; }
210 Dimensions m_dimensions;
226 static const int NumDims = internal::array_size<typename TensorEvaluator<ArgType, Device>::Dimensions>::value;
227 typedef DSizes<Index, NumDims> Dimensions; typedef in struct:Eigen::TensorEvaluator
H A DTensorConcatenation.h114 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;
116 typedef DSizes<Index, NumDims> Dimensions; typedef in struct:Eigen::TensorEvaluator
135 const Dimensions& lhs_dims = m_leftImpl.dimensions();
136 const Dimensions& rhs_dims = m_rightImpl.dimensions();
177 EIGEN_DEVICE_FUNC EIGEN_STRONG_INLINE const Dimensions& dimensions() const { return m_dimensions; }
213 const Dimensions& left_dims = m_leftImpl.dimensions();
230 const Dimensions& right_dims = m_rightImpl.dimensions();
281 Dimensions m_dimensions;
297 typedef typename Base::Dimensions Dimension typedef in struct:Eigen::TensorEvaluator
[all...]
H A DTensorGenerator.h85 typedef typename TensorEvaluator<ArgType, Device>::Dimensions Dimensions; typedef in struct:Eigen::TensorEvaluator
86 static const int NumDims = internal::array_size<Dimensions>::value;
118 EIGEN_DEVICE_FUNC EIGEN_STRONG_INLINE const Dimensions& dimensions() const { return m_dimensions; }
178 Dimensions m_dimensions;
H A DTensorArgMax.h83 typedef typename TensorEvaluator<ArgType, Device>::Dimensions Dimensions; typedef in struct:Eigen::TensorEvaluator
84 static const int NumDims = internal::array_size<Dimensions>::value;
98 EIGEN_DEVICE_FUNC EIGEN_STRONG_INLINE const Dimensions& dimensions() const {
208 typedef typename TensorEvaluator<const TensorReductionOp<ReduceOp, Dims, const TensorIndexTupleOp<ArgType> >, Device>::Dimensions Dimensions; typedef in struct:Eigen::TensorEvaluator
209 typedef typename TensorEvaluator<const TensorIndexTupleOp<ArgType> , Device>::Dimensions InputDimensions;
238 EIGEN_DEVICE_FUNC EIGEN_STRONG_INLINE const Dimensions& dimensions() const {
H A DTensorContraction.h24 template<typename Dimensions, typename LhsXprType, typename RhsXprType>
25 struct traits<TensorContractionOp<Dimensions, LhsXprType, RhsXprType> >
41 static const int NumDimensions = traits<RhsXprType>::NumDimensions + traits<RhsXprType>::NumDimensions - 2 * array_size<Dimensions>::value;
49 template<typename Dimensions, typename LhsXprType, typename RhsXprType>
50 struct eval<TensorContractionOp<Dimensions, LhsXprType, RhsXprType>, Eigen::Dense>
52 typedef const TensorContractionOp<Dimensions, LhsXprType, RhsXprType>& type;
55 template<typename Dimensions, typename LhsXprType, typename RhsXprType>
56 struct nested<TensorContractionOp<Dimensions, LhsXprType, RhsXprType>, 1, typename eval<TensorContractionOp<Dimensions, LhsXprType, RhsXprType> >::type>
58 typedef TensorContractionOp<Dimensions, LhsXprTyp
150 typedef DSizes<Index, NumDims> Dimensions; typedef in struct:Eigen::TensorContractionEvaluatorBase
610 typedef DSizes<Index, NumDims> Dimensions; typedef in struct:Eigen::TensorEvaluator
[all...]
H A DTensorLayoutSwap.h115 static const int NumDims = internal::array_size<typename TensorEvaluator<ArgType, Device>::Dimensions>::value;
116 typedef DSizes<Index, NumDims> Dimensions; typedef in struct:Eigen::TensorEvaluator
138 EIGEN_DEVICE_FUNC EIGEN_STRONG_INLINE const Dimensions& dimensions() const { return m_dimensions; }
168 Dimensions m_dimensions;
H A DTensorInflation.h82 static const int NumDims = internal::array_size<typename TensorEvaluator<ArgType, Device>::Dimensions>::value;
83 typedef DSizes<Index, NumDims> Dimensions; typedef in struct:Eigen::TensorEvaluator
112 const typename TensorEvaluator<ArgType, Device>::Dimensions& input_dims = m_impl.dimensions();
130 EIGEN_DEVICE_FUNC EIGEN_STRONG_INLINE const Dimensions& dimensions() const { return m_dimensions; }
219 Dimensions m_dimensions;
H A DTensorReverse.h106 typedef DSizes<Index, NumDims> Dimensions; typedef in struct:Eigen::TensorEvaluator
143 const Dimensions& dimensions() const { return m_dimensions; }
228 Dimensions m_dimensions;
245 typedef DSizes<Index, NumDims> Dimensions; typedef in struct:TensorEvaluator
264 const Dimensions& dimensions() const { return this->m_dimensions; }
H A DTensorDimensions.h43 template <typename Dimensions> EIGEN_DEVICE_FUNC
45 const Dimensions& dimensions)
48 dget<RowMajor ? n - 1 : (NumIndices - n), Dimensions>::value *
56 template <typename Dimensions> EIGEN_DEVICE_FUNC
57 static inline Index run(array<Index, NumIndices> const&, const Dimensions&) argument
66 template <typename Dimensions> EIGEN_DEVICE_FUNC
68 const Dimensions& dimensions)
79 template <typename Dimensions> EIGEN_DEVICE_FUNC
81 const Dimensions&)
80 run(const Index, const Dimensions&) argument
H A DTensorCustomOp.h85 typedef DSizes<Index, NumDims> Dimensions; typedef in struct:Eigen::TensorEvaluator
106 EIGEN_DEVICE_FUNC EIGEN_STRONG_INLINE const Dimensions& dimensions() const { return m_dimensions; }
150 Dimensions m_dimensions;
241 typedef DSizes<Index, NumDims> Dimensions; typedef in struct:Eigen::TensorEvaluator
262 EIGEN_DEVICE_FUNC EIGEN_STRONG_INLINE const Dimensions& dimensions() const { return m_dimensions; }
304 Dimensions m_dimensions;
H A DTensorMap.h50 typedef typename PlainObjectType::Dimensions Dimensions; typedef in class:Eigen::TensorMap
99 template <typename Dimensions>
100 EIGEN_DEVICE_FUNC EIGEN_STRONG_INLINE TensorMap(PointerArgType dataPtr, const Dimensions& dimensions)
113 EIGEN_STRONG_INLINE const Dimensions& dimensions() const { return m_dimensions; }
316 Dimensions m_dimensions;
H A DTensorStriding.h104 static const int NumDims = internal::array_size<typename TensorEvaluator<ArgType, Device>::Dimensions>::value;
105 typedef DSizes<Index, NumDims> Dimensions; typedef in struct:Eigen::TensorEvaluator
127 const typename TensorEvaluator<ArgType, Device>::Dimensions& input_dims = m_impl.dimensions();
149 EIGEN_DEVICE_FUNC EIGEN_STRONG_INLINE const Dimensions& dimensions() const { return m_dimensions; }
249 Dimensions m_dimensions;
264 static const int NumDims = internal::array_size<typename TensorEvaluator<ArgType, Device>::Dimensions>::value;
265 // typedef DSizes<Index, NumDims> Dimensions;
H A DTensorConvolution.h218 template<typename Dimensions, typename InputXprType, typename KernelXprType>
219 struct traits<TensorConvolutionOp<Dimensions, InputXprType, KernelXprType> >
240 template<typename Dimensions, typename InputXprType, typename KernelXprType>
241 struct eval<TensorConvolutionOp<Dimensions, InputXprType, KernelXprType>, Eigen::Dense>
243 typedef const TensorConvolutionOp<Dimensions, InputXprType, KernelXprType>& type;
246 template<typename Dimensions, typename InputXprType, typename KernelXprType>
247 struct nested<TensorConvolutionOp<Dimensions, InputXprType, KernelXprType>, 1, typename eval<TensorConvolutionOp<Dimensions, InputXprType, KernelXprType> >::type>
249 typedef TensorConvolutionOp<Dimensions, InputXprType, KernelXprType> type;
295 static const int NumDims = internal::array_size<typename TensorEvaluator<InputArgType, Device>::Dimensions>
298 typedef DSizes<Index, NumDims> Dimensions; typedef in struct:Eigen::TensorEvaluator
761 typedef DSizes<Index, NumDims> Dimensions; typedef in struct:Eigen::TensorEvaluator
[all...]
H A DTensorAssign.h96 typedef typename TensorEvaluator<RightArgType, Device>::Dimensions Dimensions; typedef in struct:Eigen::TensorEvaluator
113 EIGEN_DEVICE_FUNC const Dimensions& dimensions() const
/external/tensorflow/tensorflow/core/kernels/
H A Deigen_softmax_test.cc44 Tensor<float, 2>::Dimensions dims2d;
H A Dadjust_contrast_op.h52 Eigen::Tensor<int, 4>::Dimensions reshape_dims;
114 Eigen::Tensor<int, 4>::Dimensions reshape_dims;
H A Deigen_softmax.h41 typename Input::Dimensions dimensions(const Input& input) const {
/external/eigen/unsupported/test/
H A Dcxx11_tensor_concatenation.cpp50 .reshape(Tensor<int, 3>::Dimensions(2, 3, 1))
53 .concatenate(right.reshape(Tensor<int, 2>::Dimensions{{{2, 3}}}), 0);
H A Dcxx11_tensor_forced_eval.cpp58 Tensor<float, 2>::Dimensions dims2d;

Completed in 333 milliseconds

123