stored_gradient_vari.hpp

Go to the documentation of this file.

#ifndef STAN__AGRAD__REV__MATRIX__STORED_GRADIENT_HPP
#define STAN__AGRAD__REV__MATRIX__STORED_GRADIENT_HPP

#include <stan/agrad/rev/vari.hpp>

namespace stan {

  namespace agrad {

    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];
      }
    };

  }
}

#endif