stored_gradient_vari.hpp

Go to the documentation of this file.

#ifndef STAN_MATH_REV_CORE_STORED_GRADIENT_VARI_HPP
#define STAN_MATH_REV_CORE_STORED_GRADIENT_VARI_HPP
 
#include <stan/math/rev/core/vari.hpp>
 
namespace stan {
namespace math {
 
class stored_gradient_vari : public vari {
 protected:
  size_t size_;
  vari** dtrs_;
  double* partials_;
 
 public:
  stored_gradient_vari(double value, size_t size, vari** dtrs, double* partials)
      : vari(value), size_(size), dtrs_(dtrs), partials_(partials) {}
 
  void chain() {
    for (size_t i = 0; i < size_; ++i) {
      dtrs_[i]->adj_ += adj_ * partials_[i];
    }
  }
};
 
}  // namespace math
}  // namespace stan
 
#endif