partials_propagator.hpp

Go to the documentation of this file.

#ifndef STAN_MATH_FWD_META_PARTIALS_PROPOGATOR_HPP
#define STAN_MATH_FWD_META_PARTIALS_PROPOGATOR_HPP
 
#include <stan/math/prim/fun/Eigen.hpp>
#include <stan/math/prim/meta.hpp>
#include <stan/math/prim/fun/size.hpp>
#include <stan/math/fwd/core/fvar.hpp>
#include <stan/math/fwd/meta.hpp>
#include <stan/math/fwd/functor/operands_and_partials.hpp>
#include <stan/math/prim/functor/broadcast_array.hpp>
#include <stan/math/prim/functor/operands_and_partials.hpp>
#include <stan/math/prim/functor/partials_propagator.hpp>
#include <stan/math/prim/functor/apply.hpp>
#include <stan/math/prim/functor/for_each.hpp>
#include <vector>
 
namespace stan {
namespace math {
namespace internal {
 
template <typename ReturnType, typename... Ops>
class partials_propagator<ReturnType, require_fvar_t<ReturnType>, Ops...> {
 public:
  using Dx = partials_type_t<ReturnType>;
  std::tuple<
      internal::ops_partials_edge<Dx, plain_type_t<std::decay_t<Ops>>>...>
      edges_;
  using T_return_type = fvar<Dx>;
 
  template <typename... Types>
  explicit partials_propagator(Types&&... ops)
      : edges_(internal::ops_partials_edge<Dx, plain_type_t<std::decay_t<Ops>>>(
          std::forward<Types>(ops))...) {}
 
  inline T_return_type build(Dx value) {
    auto deriv = stan::math::apply(
        [](auto&... args) { return sum_dx(args...); }, edges_);
    return T_return_type(value, deriv);
  }
};
 
}  // namespace internal
}  // namespace math
}  // namespace stan
#endif