math/opencl_2prim_2constraint_2unit__vector__constrain_8hpp_source.html

#ifndef STAN_MATH_OPENCL_PRIM_CONSTRAINT_UNIT_VECTOR_CONSTRAIN_BLOCK_HPP

#define STAN_MATH_OPENCL_PRIM_CONSTRAINT_UNIT_VECTOR_CONSTRAIN_BLOCK_HPP

#ifdef STAN_OPENCL


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

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


namespace stan {

namespace math {


template <typename T_x,

          require_all_kernel_expressions_and_none_scalar_t<T_x>* = nullptr>

inline auto unit_vector_constrain(T_x&& x) {

  using std::sqrt;

  check_nonzero_size("unit_vector_constrain", "x", x);

  value_type_t<T_x> SN = dot_self(x);

  check_positive_finite("unit_vector_constrain", "norm", SN);

  return elt_divide(x, sqrt(SN));

}


template <typename T_x,

          require_all_kernel_expressions_and_none_scalar_t<T_x>* = nullptr>

inline auto unit_vector_constrain(T_x&& x, double& lp) {

  using std::sqrt;

  check_nonzero_size("unit_vector_constrain", "x", x);

  value_type_t<T_x> SN = dot_self(x);

  check_positive_finite("unit_vector_constrain", "norm", SN);

  lp -= 0.5 * SN;

  return elt_divide(x, sqrt(SN));

}


}  // namespace math

}  // namespace stan

#endif

#endif

stan::math::elt_divide
elt_divide_< as_operation_cl_t< T_a >, as_operation_cl_t< T_b > > elt_divide(T_a &&a, T_b &&b)
Definition binary_operation.hpp:209

stan::require_all_kernel_expressions_and_none_scalar_t
require_all_t< is_kernel_expression_and_not_scalar< Types >... > require_all_kernel_expressions_and_none_scalar_t
Enables a template if all given types are non-scalar types that are a valid kernel generator expressi...
Definition is_kernel_expression.hpp:58

stan::value_type_t
typename value_type< T >::type value_type_t
Helper function for accessing underlying type.
Definition value_type.hpp:35

kernel_generator.hpp

matrix_cl.hpp

stan::math::unit_vector_constrain
auto unit_vector_constrain(const EigMat &y)
Definition unit_vector_constrain.hpp:20

stan::math::sqrt
fvar< T > sqrt(const fvar< T > &x)
Definition sqrt.hpp:17

stan::math::check_nonzero_size
void check_nonzero_size(const char *function, const char *name, const T_y &y)
Check if the specified matrix/vector is of non-zero size.
Definition check_nonzero_size.hpp:22

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

stan::math::check_positive_finite
void check_positive_finite(const char *function, const char *name, const T_y &y)
Check if y is positive and finite.
Definition check_positive_finite.hpp:22

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