![]() |
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::identity (__global double *A, unsigned int rows, unsigned int cols) |
Makes an identity matrix on the OpenCL device. More... | |
__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. More... | |
Variables | |
const kernel_cl< out_buffer, int, int > | stan::math::opencl_kernels::identity ("identity", {indexing_helpers, identity_kernel_code}) |
See the docs for identity() . More... | |
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() . More... | |