Stan Math Library
5.0.0
Automatic Differentiation
|
#include <stan/math/opencl/kernel_cl.hpp>
#include <stan/math/opencl/buffer_types.hpp>
#include <stan/math/opencl/matrix_cl_view.hpp>
#include <string>
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. | |
namespace | stan::math::opencl_kernels |
Functions | |
__kernel void | stan::math::opencl_kernels::add_batch (__global double *B, __global double *A, unsigned int rows, unsigned int cols, unsigned int batch_size) |
Sums a batch of matrices. | |
Variables | |
const kernel_cl< out_buffer, in_buffer, int, int, int > | stan::math::opencl_kernels::add_batch ("add_batch", {indexing_helpers, add_batch_kernel_code}) |
See the docs for add_batch() . | |