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

◆ categorical_logit_glm_beta_derivative()

__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.

Parameters
[in,out]beta_derivativederivative wrt beta
temptemporary workspace of size global_size*N_classes
[in]ya scalar or vector of classes
[in]xdesign matrix
N_instancesnumber of instances
N_attributesnumber of attributes
N_classesnumber of classes
is_y_vector0 or 1 - whether y is a vector (alternatively it is a scalar)

Definition at line 174 of file categorical_logit_glm_lpmf.hpp.