math/opencl_2prim_2symmetrize__from__upper__tri_8hpp_source.html

#ifndef STAN_MATH_OPENCL_PRIM_SYMMETRIZE_FROM_UPPER_TRI_HPP

#define STAN_MATH_OPENCL_PRIM_SYMMETRIZE_FROM_UPPER_TRI_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 symmetrize_from_upper_tri(T_x&& x) {

  check_square("symmetrize_from_upper_tri", "x", x);

  return make_holder_cl(

      [](auto& arg) {

        return select(col_index() < row_index(), transpose(arg), arg);

      },

      std::forward<T_x>(x));

}

}  // namespace math

}  // namespace stan

#endif

#endif

stan::math::col_index
Represents operation that determines column index.
Definition index.hpp:80

stan::math::row_index
Represents operation that determines row index.
Definition index.hpp:17

stan::math::make_holder_cl
auto make_holder_cl(const T &func, Args &&... args)
Constructs an expression from given arguments using given functor.
Definition holder_cl.hpp:125

stan::math::select
select_< as_operation_cl_t< T_condition >, as_operation_cl_t< T_then >, as_operation_cl_t< T_else > > select(T_condition &&condition, T_then &&then, T_else &&els)
Selection operation on kernel generator expressions.
Definition select.hpp:148

stan::math::transpose
auto transpose(Arg &&a)
Transposes a kernel generator expression.
Definition transpose.hpp:139

kernel_generator.hpp

matrix_cl.hpp

stan::math::check_square
void check_square(const char *function, const char *name, const T_y &y)
Check if the specified matrix is square.
Definition check_square.hpp:23

stan::math::arg
fvar< T > arg(const std::complex< fvar< T > > &z)
Return the phase angle of the complex argument.
Definition arg.hpp:19

stan::math::symmetrize_from_upper_tri
auto symmetrize_from_upper_tri(T_x &&x)
Return a symmetric matrix using elements from the lower triangular part of the input matrix.
Definition symmetrize_from_upper_tri.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