![]() |
Stan Math Library
2.20.0
reverse mode automatic differentiation
|
Go to the source code of this file.
Namespaces | |
stan | |
stan::math | |
Matrices and templated mathematical functions. | |
stan::math::opencl_kernels | |
Functions | |
__kernel void | stan::math::opencl_kernels::add (__global double *C, __global double *A, __global double *B, unsigned int rows, unsigned int cols) |
Matrix addition on the OpenCL device. More... | |
__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. More... | |
Variables | |
const kernel_cl< out_buffer, in_buffer, in_buffer, int, int > | stan::math::opencl_kernels::add ("add", {indexing_helpers, add_kernel_code}) |
See the docs for add() . More... | |
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() . More... | |