precomp_vv_vari.hpp

Go to the documentation of this file.

#ifndef STAN_MATH_REV_CORE_PRECOMP_VV_VARI_HPP
#define STAN_MATH_REV_CORE_PRECOMP_VV_VARI_HPP
 
#include <stan/math/rev/core/vari.hpp>
#include <stan/math/rev/core/vv_vari.hpp>
 
namespace stan {
namespace math {
 
// use for single precomputed partials
class precomp_vv_vari final : public op_vv_vari {
 protected:
  double da_;
  double db_;
 
 public:
  precomp_vv_vari(double val, vari* avi, vari* bvi, double da, double db)
      : op_vv_vari(val, avi, bvi), da_(da), db_(db) {}
  void chain() {
    avi_->adj_ += adj_ * da_;
    bvi_->adj_ += adj_ * db_;
  }
};
 
}  // namespace math
}  // namespace stan
#endif