Stan Math Library
4.9.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.