![]() |
Stan Math Library
5.1.0
Automatic Differentiation
|
#include <stan/math/opencl/kernel_cl.hpp>#include <stan/math/opencl/buffer_types.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::batch_identity (__global double *A, unsigned int batch_rows, unsigned int size) |
| Makes a batch of smaller identity matrices inside the input matrix. | |
Variables | |
| const kernel_cl< out_buffer, int, int > | stan::math::opencl_kernels::batch_identity ("batch_identity", {indexing_helpers, batch_identity_kernel_code}) |
| See the docs for batch_identity() . | |