math/opencl_2rev_2dot__self_8hpp_source.html

#ifndef STAN_MATH_OPENCL_REV_DOT_SELF_HPP

#define STAN_MATH_OPENCL_REV_DOT_SELF_HPP

#ifdef STAN_OPENCL


#include <stan/math/opencl/kernel_generator.hpp>

#include <stan/math/opencl/prim/dot_self.hpp>

#include <stan/math/rev/core.hpp>


namespace stan {

namespace math {


template <typename T,

          require_all_kernel_expressions_and_none_scalar_t<T>* = nullptr>

inline var dot_self(const var_value<T>& v) {

  return make_callback_var(dot_self(v.val()), [v](vari& res) mutable {

    v.adj() += 2.0 * res.adj() * v.val();

  });

}


}  // namespace math

}  // namespace stan


#endif

#endif

stan::math::var_value< double >

stan::math::vari_value
Definition vari.hpp:17

kernel_generator.hpp

stan::math::make_callback_var
var_value< plain_type_t< T > > make_callback_var(T &&value, F &&functor)
Creates a new var initialized with a callback_vari with a given value and reverse-pass callback funct...
Definition callback_vari.hpp:61

stan::math::dot_self
auto dot_self(const T &a)
Returns squared norm of a vector or matrix.
Definition dot_self.hpp:21

stan
The lgamma implementation in stan-math is based on either the reentrant safe lgamma_r implementation ...
Definition unit_vector_constrain.hpp:15

dot_self.hpp

core.hpp