![]() |
Stan Math Library
5.1.0
Automatic Differentiation
|
| __kernel void stan::math::opencl_kernels::multiply_transpose | ( | const __global double * | A, |
| __global double * | B, | ||
| const int | M, | ||
| const int | N | ||
| ) |
Matrix multiplication of the form A*A^T on the OpenCL device.
| [in] | A | matrix A |
| [out] | B | the output matrix |
| [in] | M | Number of rows for matrix A |
| [in] | N | Number of cols for matrix A and the number of rows for matrix A^T |
Definition at line 24 of file multiply_transpose.hpp.