Automatic Differentiation
 
Loading...
Searching...
No Matches
count_vars.hpp File Reference
#include <stan/math/prim/meta.hpp>
#include <stan/math/rev/meta.hpp>
#include <stan/math/rev/core/var.hpp>
#include <utility>
#include <vector>

Go to the source code of this file.

Namespaces

namespace  stan
 The lgamma implementation in stan-math is based on either the reentrant safe lgamma_r implementation from C or the boost::math::lgamma implementation.
 
namespace  stan::math
 Matrices and templated mathematical functions.
 
namespace  stan::math::internal
 A comparator that works for any container type that has the brackets operator.
 

Functions

template<typename VecVar , require_std_vector_vt< is_var, VecVar > * = nullptr, typename... Pargs>
size_t stan::math::internal::count_vars_impl (size_t count, VecVar &&x, Pargs &&... args)
 Count the number of vars in x (a std::vector of vars), add it to the running total, count the number of vars in the remaining arguments and return the result.
 
template<typename VecContainer , require_std_vector_st< is_var, VecContainer > * = nullptr, require_std_vector_vt< is_container, VecContainer > * = nullptr, typename... Pargs>
size_t stan::math::internal::count_vars_impl (size_t count, VecContainer &&x, Pargs &&... args)
 Count the number of vars in x (a std::vector holding other containers), add it to the running total, count the number of vars in the remaining arguments and return the result.
 
template<typename EigT , require_eigen_vt< is_var, EigT > * = nullptr, typename... Pargs>
size_t stan::math::internal::count_vars_impl (size_t count, EigT &&x, Pargs &&... args)
 Count the number of vars in x (an eigen container), add it to the running total, count the number of vars in the remaining arguments and return the result.
 
template<typename... Pargs>
size_t stan::math::internal::count_vars_impl (size_t count, const var &x, Pargs &&... args)
 Add one to the running total number of vars, count the number of vars in the remaining arguments and return the result.
 
template<typename Arith , require_arithmetic_t< scalar_type_t< Arith > > * = nullptr, typename... Pargs>
size_t stan::math::internal::count_vars_impl (size_t count, Arith &x, Pargs &&... args)
 Arguments without vars contribute zero to the total number of vars.
 
size_t stan::math::internal::count_vars_impl (size_t count)
 End count_vars_impl recursion and return total number of counted vars.
 
template<typename... Pargs>
size_t stan::math::count_vars (Pargs &&... args)
 Count the number of vars in the input argument list.