hypot.hpp

Go to the documentation of this file.

#ifndef STAN_MATH_REV_FUN_HYPOT_HPP
#define STAN_MATH_REV_FUN_HYPOT_HPP
 
#include <stan/math/rev/meta.hpp>
#include <stan/math/rev/core.hpp>
#include <stan/math/prim/fun/hypot.hpp>
 
namespace stan {
namespace math {
 
inline var hypot(const var& a, const var& b) {
  return make_callback_var(hypot(a.val(), b.val()), [a, b](auto& vi) mutable {
    a.adj() += vi.adj() * a.val() / vi.val();
    b.adj() += vi.adj() * b.val() / vi.val();
  });
}
 
inline var hypot(const var& a, double b) {
  return make_callback_var(hypot(a.val(), b), [a](auto& vi) mutable {
    a.adj() += vi.adj() * a.val() / vi.val();
  });
}
 
inline var hypot(double a, const var& b) {
  return make_callback_var(hypot(b.val(), a), [b](auto& vi) mutable {
    b.adj() += vi.adj() * b.val() / vi.val();
  });
}
 
}  // namespace math
}  // namespace stan
#endif