![]() |
Stan Math Library
5.1.0
Automatic Differentiation
|
| __kernel void stan::math::opencl_kernels::matrix_multiply | ( | const __global double * | A, |
| const __global double * | B, | ||
| __global double * | C, | ||
| const int | M, | ||
| const int | N, | ||
| const int | K, | ||
| unsigned int | view_A, | ||
| unsigned int | view_B | ||
| ) |
Matrix multiplication on the OpenCL device.
| [in] | A | the left matrix in matrix multiplication |
| [in] | B | the right matrix in matrix multiplication |
| [out] | C | the output matrix |
| [in] | M | Number of rows for matrix A |
| [in] | N | Number of cols for matrix B |
| [in] | K | Number of cols for matrix A and number of rows for matrix B |
| [in] | view_A | the triangularity of A (lower, upper or none) |
| [in] | view_B | the triangularity of B (lower, upper or none) |
Definition at line 28 of file matrix_multiply.hpp.