![]() |
Stan Math Library
5.1.0
Automatic Differentiation
|
| __kernel void stan::math::opencl_kernels::categorical_logit_glm_beta_derivative | ( | __global double * | beta_derivative, |
| __global double * | temp, | ||
| const __global int * | y, | ||
| const __global double * | x, | ||
| const int | N_instances, | ||
| const int | N_attributes, | ||
| const int | N_classes, | ||
| const int | is_y_vector | ||
| ) |
Calculates derivative wrt beta.
Must be run with global size of local_size*N_attributes.
| [in,out] | beta_derivative | derivative wrt beta |
| temp | temporary workspace of size global_size*N_classes | |
| [in] | y | a scalar or vector of classes |
| [in] | x | design matrix |
| N_instances | number of instances | |
| N_attributes | number of attributes | |
| N_classes | number of classes | |
| is_y_vector | 0 or 1 - whether y is a vector (alternatively it is a scalar) |
Definition at line 174 of file categorical_logit_glm_lpmf.hpp.