Searched defs:weights_gpu (Results 1 – 2 of 2) sorted by relevance
| /dports/misc/darknet/darknet-darknet_yolo_v4_pre/src/ |
| H A D | blas_kernels.cu | 439 __global__ void constrain_weight_updates_kernel(int N, float coef, float *weights_gpu, float *weigh… in constrain_weight_updates_kernel()
451 extern "C" void constrain_weight_updates_ongpu(int N, float coef, float *weights_gpu, float *weight… in constrain_weight_updates_ongpu()
912 …s_of_layers_gpu, float **layers_output_gpu, float *out, float *in, float *weights_gpu, int nweight… in shortcut_singlelayer_simple_kernel()
934 …s_of_layers_gpu, float **layers_output_gpu, float *out, float *in, float *weights_gpu, int nweight… in shortcut_multilayer_kernel()
1001 …s_of_layers_gpu, float **layers_output_gpu, float *out, float *in, float *weights_gpu, int nweight… in shortcut_multilayer_gpu()
1016 …float **layers_delta_gpu, float *delta_out, float *delta_in, float *weights_gpu, float *weight_upd… in backward_shortcut_multilayer_kernel()
1127 …float **layers_delta_gpu, float *delta_out, float *delta_in, float *weights_gpu, float *weight_upd… in backward_shortcut_multilayer_gpu()
|
| /dports/misc/darknet/darknet-darknet_yolo_v4_pre/include/ |
| H A D | darknet.h | 583 float * weights_gpu; member
|