Go to the source code of this file.
|
namespace | stan |
| The lgamma implementation in stan-math is based on either the reentrant safe lgamma_r implementation from C or the boost::math::lgamma implementation.
|
|
namespace | stan::math |
| Matrices and templated mathematical functions.
|
|
namespace | stan::math::opencl_kernels |
|
|
__kernel void | stan::math::opencl_kernels::categorical_logit_glm (__global double *logp_global, __global double *exp_lin_global, __global double *inv_sum_exp_lin_global, __global double *neg_softmax_lin_global, __global double *alpha_derivative, const __global int *y_global, const __global double *x_beta_global, const __global double *alpha_global, const int N_instances, const int N_attributes, const int N_classes, const int is_y_vector, const int need_alpha_derivative, const int need_neg_softmax_lin_global) |
| GPU implementation of Generalized Linear Model (GLM) with categorical distribution and logit (softmax) link function.
|
|
__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.
|
|
|
const kernel_cl< out_buffer, out_buffer, out_buffer, out_buffer, out_buffer, in_buffer, in_buffer, in_buffer, int, int, int, int, int, int > | stan::math::opencl_kernels::categorical_logit_glm ("categorical_logit_glm", {categorical_logit_glm_kernel_code}, {{"REDUCTION_STEP_SIZE", 4}, {"LOCAL_SIZE_", 64}}) |
| See the docs for categorical_logit_glm() .
|
|
const kernel_cl< in_out_buffer, in_out_buffer, in_buffer, in_buffer, int, int, int, int > | stan::math::opencl_kernels::categorical_logit_glm_beta_derivative ("categorical_logit_glm_beta_derivative", {categorical_logit_glm_beta_derivative_kernel_code}) |
| See the docs for categorical_logit_glm_beta_derivative() .
|
|