Stan Math Library
4.9.0
Automatic Differentiation
|
#include <stan/math/opencl/rev/vari.hpp>
#include <stan/math/opencl/rev/arena_type.hpp>
#include <stan/math/opencl/rev/to_arena.hpp>
#include <stan/math/opencl/copy.hpp>
#include <stan/math/rev/core.hpp>
#include <stan/math/rev/meta.hpp>
#include <stan/math/prim/err.hpp>
#include <stan/math/prim/fun/Eigen.hpp>
#include <stan/math/prim/fun/vec_concat.hpp>
#include <CL/opencl.hpp>
#include <iostream>
#include <vector>
#include <algorithm>
#include <type_traits>
Go to the source code of this file.
Namespaces | |
namespace | stan |
The lgamma implementation in stan-math is based on either the reentrant safe lgamma_r implementation from C or the boost::math::lgamma implementation. | |
namespace | stan::math |
Matrices and templated mathematical functions. | |
Functions | |
template<typename T > | |
var_value< matrix_cl< value_type_t< T > > > | stan::math::to_matrix_cl (const var_value< T > &a) |
Copies the source var containing Eigen matrices to destination var that has data stored on the OpenCL device. | |
template<typename T , require_stan_scalar_t< T > * = nullptr> | |
var_value< matrix_cl< value_type_t< T > > > | stan::math::to_matrix_cl (const std::vector< var_value< T > > &a) |
Copies the source std::vector of vars to a destination var that has data stored on the OpenCL device. | |
template<typename T , require_eigen_vt< is_var, T > * = nullptr> | |
var_value< matrix_cl< value_type_t< value_type_t< T > > > > | stan::math::to_matrix_cl (const T &src) |
Copies the source Eigen matrix of vars to the destination matrix that is stored on the OpenCL device. | |
template<typename T , require_eigen_vt< is_var, T > * = nullptr> | |
var_value< matrix_cl< value_type_t< value_type_t< T > > > > | stan::math::to_matrix_cl (const std::vector< T > &src) |
Copies the source vector of Eigen matrices of vars to the destination matrix that is stored on the OpenCL device. | |
template<typename T_dst , typename T , require_var_vt< is_eigen, T_dst > * = nullptr, require_all_kernel_expressions_t< T > * = nullptr> | |
T_dst | stan::math::from_matrix_cl (const var_value< T > &a) |
Copies the source var that has data stored on the OpenCL device to destination var containing Eigen matrix. | |
template<typename T_dst , typename T , require_std_vector_vt< is_var, T_dst > * = nullptr, require_all_stan_scalar_t< value_type_t< T_dst > > * = nullptr, require_all_kernel_expressions_t< T > * = nullptr> | |
T_dst | stan::math::from_matrix_cl (const var_value< T > &a) |
Copies the source var that has data stored on the OpenCL device to destination std::vector containing vars. | |
template<typename T , require_all_kernel_expressions_t< T > * = nullptr> | |
auto | stan::math::from_matrix_cl (const var_value< T > &src) |
Copies the source var that has data stored on the OpenCL device to destination Eigen matrix containing vars. | |