fill.hpp

Go to the documentation of this file.

#ifndef STAN_MATH_REV_FUN_FILL_HPP
#define STAN_MATH_REV_FUN_FILL_HPP
 
#include <stan/math/prim/fun/Eigen.hpp>
#include <stan/math/rev/meta.hpp>
#include <stan/math/rev/core.hpp>
#include <stan/math/prim/fun/fill.hpp>
#include <vector>
 
namespace stan {
namespace math {
 
template <typename VarMat, typename S, require_var_matrix_t<VarMat>* = nullptr,
          require_var_t<S>* = nullptr>
inline void fill(VarMat& x, const S& y) {
  arena_t<plain_type_t<value_type_t<VarMat>>> prev_vals(x.val().eval());
  x.vi_->val_.fill(y.val());
  reverse_pass_callback([x, y, prev_vals]() mutable {
    x.vi_->val_ = prev_vals;
    y.adj() += x.adj().sum();
    x.adj().setZero();
  });
}
 
template <typename VarMat, typename S, require_var_matrix_t<VarMat>* = nullptr,
          require_arithmetic_t<S>* = nullptr>
inline void fill(VarMat& x, const S& y) {
  arena_t<plain_type_t<value_type_t<VarMat>>> prev_vals(x.val().eval());
  x.vi_->val_.fill(y);
  reverse_pass_callback([x, prev_vals]() mutable {
    x.vi_->val_ = prev_vals;
    x.adj().setZero();
  });
}
 
}  // namespace math
}  // namespace stan
 
#endif