Automatic Differentiation
 
Loading...
Searching...
No Matches

◆ ordered_logistic_glm()

__kernel void stan::math::opencl_kernels::ordered_logistic_glm ( __global double *  location_sum,
__global double *  logp_global,
__global double *  location_derivative,
__global double *  cuts_derivative,
const __global int *  y_global,
const __global double *  x,
const __global double *  beta,
const __global double *  cuts,
const int  N_instances,
const int  N_attributes,
const int  N_classes,
const int  is_y_vector,
const int  need_location_derivative,
const int  need_cuts_derivative 
)

GPU implementation of ordinal regression Generalized Linear Model (GLM).

Must be run with at least N_instances threads and local size equal to LOCAL_SIZE_.

Parameters
[out]location_sumpartially summed location (1 value per work group)
[out]logp_globalpartially summed log probability (1 value per work group)
[out]location_derivativederivative wrt location
[out]cuts_derivativepartially summed derivative wrt cuts (1 column per work group)
[in]y_globala scalar or vector of classes.
[in]xdesign matrix
[in]betaweight vector
[in]cutscutpoints vector
N_instancesnumber of cases
N_attributesnumber of attributes
N_classesnumber of classes
is_y_vector0 or 1 - whether y is a vector (alternatively it is a scalar)
need_location_derivativeinterpreted as boolean - whether location_derivative needs to be computed
need_cuts_derivativeinterpreted as boolean - whether cuts_derivative needs to be computed

Definition at line 43 of file ordered_logistic_glm_lpmf.hpp.