diff options
Diffstat (limited to 'runtimes/nn/depend/external/eigen/Eigen/src/Core/Visitor.h')
-rw-r--r-- | runtimes/nn/depend/external/eigen/Eigen/src/Core/Visitor.h | 273 |
1 files changed, 0 insertions, 273 deletions
diff --git a/runtimes/nn/depend/external/eigen/Eigen/src/Core/Visitor.h b/runtimes/nn/depend/external/eigen/Eigen/src/Core/Visitor.h deleted file mode 100644 index 54c1883d9..000000000 --- a/runtimes/nn/depend/external/eigen/Eigen/src/Core/Visitor.h +++ /dev/null @@ -1,273 +0,0 @@ -// This file is part of Eigen, a lightweight C++ template library -// for linear algebra. -// -// Copyright (C) 2008 Gael Guennebaud <gael.guennebaud@inria.fr> -// -// This Source Code Form is subject to the terms of the Mozilla -// Public License v. 2.0. If a copy of the MPL was not distributed -// with this file, You can obtain one at http://mozilla.org/MPL/2.0/. - -#ifndef EIGEN_VISITOR_H -#define EIGEN_VISITOR_H - -namespace Eigen { - -namespace internal { - -template<typename Visitor, typename Derived, int UnrollCount> -struct visitor_impl -{ - enum { - col = (UnrollCount-1) / Derived::RowsAtCompileTime, - row = (UnrollCount-1) % Derived::RowsAtCompileTime - }; - - EIGEN_DEVICE_FUNC - static inline void run(const Derived &mat, Visitor& visitor) - { - visitor_impl<Visitor, Derived, UnrollCount-1>::run(mat, visitor); - visitor(mat.coeff(row, col), row, col); - } -}; - -template<typename Visitor, typename Derived> -struct visitor_impl<Visitor, Derived, 1> -{ - EIGEN_DEVICE_FUNC - static inline void run(const Derived &mat, Visitor& visitor) - { - return visitor.init(mat.coeff(0, 0), 0, 0); - } -}; - -template<typename Visitor, typename Derived> -struct visitor_impl<Visitor, Derived, Dynamic> -{ - EIGEN_DEVICE_FUNC - static inline void run(const Derived& mat, Visitor& visitor) - { - visitor.init(mat.coeff(0,0), 0, 0); - for(Index i = 1; i < mat.rows(); ++i) - visitor(mat.coeff(i, 0), i, 0); - for(Index j = 1; j < mat.cols(); ++j) - for(Index i = 0; i < mat.rows(); ++i) - visitor(mat.coeff(i, j), i, j); - } -}; - -// evaluator adaptor -template<typename XprType> -class visitor_evaluator -{ -public: - EIGEN_DEVICE_FUNC - explicit visitor_evaluator(const XprType &xpr) : m_evaluator(xpr), m_xpr(xpr) {} - - typedef typename XprType::Scalar Scalar; - typedef typename XprType::CoeffReturnType CoeffReturnType; - - enum { - RowsAtCompileTime = XprType::RowsAtCompileTime, - CoeffReadCost = internal::evaluator<XprType>::CoeffReadCost - }; - - EIGEN_DEVICE_FUNC Index rows() const { return m_xpr.rows(); } - EIGEN_DEVICE_FUNC Index cols() const { return m_xpr.cols(); } - EIGEN_DEVICE_FUNC Index size() const { return m_xpr.size(); } - - EIGEN_DEVICE_FUNC CoeffReturnType coeff(Index row, Index col) const - { return m_evaluator.coeff(row, col); } - -protected: - internal::evaluator<XprType> m_evaluator; - const XprType &m_xpr; -}; -} // end namespace internal - -/** Applies the visitor \a visitor to the whole coefficients of the matrix or vector. - * - * The template parameter \a Visitor is the type of the visitor and provides the following interface: - * \code - * struct MyVisitor { - * // called for the first coefficient - * void init(const Scalar& value, Index i, Index j); - * // called for all other coefficients - * void operator() (const Scalar& value, Index i, Index j); - * }; - * \endcode - * - * \note compared to one or two \em for \em loops, visitors offer automatic - * unrolling for small fixed size matrix. - * - * \sa minCoeff(Index*,Index*), maxCoeff(Index*,Index*), DenseBase::redux() - */ -template<typename Derived> -template<typename Visitor> -EIGEN_DEVICE_FUNC -void DenseBase<Derived>::visit(Visitor& visitor) const -{ - typedef typename internal::visitor_evaluator<Derived> ThisEvaluator; - ThisEvaluator thisEval(derived()); - - enum { - unroll = SizeAtCompileTime != Dynamic - && SizeAtCompileTime * ThisEvaluator::CoeffReadCost + (SizeAtCompileTime-1) * internal::functor_traits<Visitor>::Cost <= EIGEN_UNROLLING_LIMIT - }; - return internal::visitor_impl<Visitor, ThisEvaluator, unroll ? int(SizeAtCompileTime) : Dynamic>::run(thisEval, visitor); -} - -namespace internal { - -/** \internal - * \brief Base class to implement min and max visitors - */ -template <typename Derived> -struct coeff_visitor -{ - typedef typename Derived::Scalar Scalar; - Index row, col; - Scalar res; - EIGEN_DEVICE_FUNC - inline void init(const Scalar& value, Index i, Index j) - { - res = value; - row = i; - col = j; - } -}; - -/** \internal - * \brief Visitor computing the min coefficient with its value and coordinates - * - * \sa DenseBase::minCoeff(Index*, Index*) - */ -template <typename Derived> -struct min_coeff_visitor : coeff_visitor<Derived> -{ - typedef typename Derived::Scalar Scalar; - EIGEN_DEVICE_FUNC - void operator() (const Scalar& value, Index i, Index j) - { - if(value < this->res) - { - this->res = value; - this->row = i; - this->col = j; - } - } -}; - -template<typename Scalar> -struct functor_traits<min_coeff_visitor<Scalar> > { - enum { - Cost = NumTraits<Scalar>::AddCost - }; -}; - -/** \internal - * \brief Visitor computing the max coefficient with its value and coordinates - * - * \sa DenseBase::maxCoeff(Index*, Index*) - */ -template <typename Derived> -struct max_coeff_visitor : coeff_visitor<Derived> -{ - typedef typename Derived::Scalar Scalar; - EIGEN_DEVICE_FUNC - void operator() (const Scalar& value, Index i, Index j) - { - if(value > this->res) - { - this->res = value; - this->row = i; - this->col = j; - } - } -}; - -template<typename Scalar> -struct functor_traits<max_coeff_visitor<Scalar> > { - enum { - Cost = NumTraits<Scalar>::AddCost - }; -}; - -} // end namespace internal - -/** \fn DenseBase<Derived>::minCoeff(IndexType* rowId, IndexType* colId) const - * \returns the minimum of all coefficients of *this and puts in *row and *col its location. - * \warning the result is undefined if \c *this contains NaN. - * - * \sa DenseBase::minCoeff(Index*), DenseBase::maxCoeff(Index*,Index*), DenseBase::visit(), DenseBase::minCoeff() - */ -template<typename Derived> -template<typename IndexType> -EIGEN_DEVICE_FUNC -typename internal::traits<Derived>::Scalar -DenseBase<Derived>::minCoeff(IndexType* rowId, IndexType* colId) const -{ - internal::min_coeff_visitor<Derived> minVisitor; - this->visit(minVisitor); - *rowId = minVisitor.row; - if (colId) *colId = minVisitor.col; - return minVisitor.res; -} - -/** \returns the minimum of all coefficients of *this and puts in *index its location. - * \warning the result is undefined if \c *this contains NaN. - * - * \sa DenseBase::minCoeff(IndexType*,IndexType*), DenseBase::maxCoeff(IndexType*,IndexType*), DenseBase::visit(), DenseBase::minCoeff() - */ -template<typename Derived> -template<typename IndexType> -EIGEN_DEVICE_FUNC -typename internal::traits<Derived>::Scalar -DenseBase<Derived>::minCoeff(IndexType* index) const -{ - EIGEN_STATIC_ASSERT_VECTOR_ONLY(Derived) - internal::min_coeff_visitor<Derived> minVisitor; - this->visit(minVisitor); - *index = IndexType((RowsAtCompileTime==1) ? minVisitor.col : minVisitor.row); - return minVisitor.res; -} - -/** \fn DenseBase<Derived>::maxCoeff(IndexType* rowId, IndexType* colId) const - * \returns the maximum of all coefficients of *this and puts in *row and *col its location. - * \warning the result is undefined if \c *this contains NaN. - * - * \sa DenseBase::minCoeff(IndexType*,IndexType*), DenseBase::visit(), DenseBase::maxCoeff() - */ -template<typename Derived> -template<typename IndexType> -EIGEN_DEVICE_FUNC -typename internal::traits<Derived>::Scalar -DenseBase<Derived>::maxCoeff(IndexType* rowPtr, IndexType* colPtr) const -{ - internal::max_coeff_visitor<Derived> maxVisitor; - this->visit(maxVisitor); - *rowPtr = maxVisitor.row; - if (colPtr) *colPtr = maxVisitor.col; - return maxVisitor.res; -} - -/** \returns the maximum of all coefficients of *this and puts in *index its location. - * \warning the result is undefined if \c *this contains NaN. - * - * \sa DenseBase::maxCoeff(IndexType*,IndexType*), DenseBase::minCoeff(IndexType*,IndexType*), DenseBase::visitor(), DenseBase::maxCoeff() - */ -template<typename Derived> -template<typename IndexType> -EIGEN_DEVICE_FUNC -typename internal::traits<Derived>::Scalar -DenseBase<Derived>::maxCoeff(IndexType* index) const -{ - EIGEN_STATIC_ASSERT_VECTOR_ONLY(Derived) - internal::max_coeff_visitor<Derived> maxVisitor; - this->visit(maxVisitor); - *index = (RowsAtCompileTime==1) ? maxVisitor.col : maxVisitor.row; - return maxVisitor.res; -} - -} // end namespace Eigen - -#endif // EIGEN_VISITOR_H |