hypergeometric_2F1.hpp

Go to the documentation of this file.

#ifndef STAN_MATH_REV_FUN_HYPERGEOMETRIC_2F1_HPP
#define STAN_MATH_REV_FUN_HYPERGEOMETRIC_2F1_HPP
 
#include <stan/math/rev/core.hpp>
#include <stan/math/rev/meta.hpp>
#include <stan/math/prim/fun/grad_2F1.hpp>
#include <stan/math/prim/fun/hypergeometric_2F1.hpp>
 
namespace stan {
namespace math {
 
template <typename Ta1, typename Ta2, typename Tb, typename Tz,
          require_all_stan_scalar_t<Ta1, Ta2, Tb, Tz>* = nullptr,
          require_any_var_t<Ta1, Ta2, Tb, Tz>* = nullptr>
inline return_type_t<Ta1, Ta2, Tb, Tz> hypergeometric_2F1(const Ta1& a1,
                                                          const Ta2& a2,
                                                          const Tb& b,
                                                          const Tz& z) {
  double a1_dbl = value_of(a1);
  double a2_dbl = value_of(a2);
  double b_dbl = value_of(b);
  double z_dbl = value_of(z);
 
  return make_callback_var(hypergeometric_2F1(a1_dbl, a2_dbl, b_dbl, z_dbl),
                           [a1, a2, b, z](auto& vi) mutable {
                             auto grad_tuple = grad_2F1(a1, a2, b, z);
 
                             if constexpr (is_autodiff_v<Ta1>) {
                               a1.adj() += vi.adj() * std::get<0>(grad_tuple);
                             }
                             if constexpr (is_autodiff_v<Ta2>) {
                               a2.adj() += vi.adj() * std::get<1>(grad_tuple);
                             }
                             if constexpr (is_autodiff_v<Tb>) {
                               b.adj() += vi.adj() * std::get<2>(grad_tuple);
                             }
                             if constexpr (is_autodiff_v<Tz>) {
                               z.adj() += vi.adj() * std::get<3>(grad_tuple);
                             }
                           });
}
}  // namespace math
}  // namespace stan
#endif