Searched refs:r_gpu (Results 1 – 3 of 3) sorted by relevance
| /dports/misc/darknet/darknet-darknet_yolo_v4_pre/src/ |
| H A D | gru_layer.c | 103 l.r_gpu = cuda_make_array(l.output_gpu, batch*outputs); in make_gru_layer()
259 copy_ongpu(l.outputs*l.batch, input_r_layer.output_gpu, 1, l.r_gpu, 1); in forward_gru_layer_gpu()
260 axpy_ongpu(l.outputs*l.batch, 1, state_r_layer.output_gpu, 1, l.r_gpu, 1); in forward_gru_layer_gpu()
263 activate_array_ongpu(l.r_gpu, l.outputs*l.batch, LOGISTIC); in forward_gru_layer_gpu()
266 mul_ongpu(l.outputs*l.batch, l.r_gpu, 1, l.forgot_state_gpu, 1); in forward_gru_layer_gpu()
329 copy_ongpu(l.outputs*l.batch, input_r_layer.output_gpu, 1, l.r_gpu, 1); in backward_gru_layer_gpu()
330 axpy_ongpu(l.outputs*l.batch, 1, state_r_layer.output_gpu, 1, l.r_gpu, 1); in backward_gru_layer_gpu()
333 activate_array_ongpu(l.r_gpu, l.outputs*l.batch, LOGISTIC); in backward_gru_layer_gpu()
355 mul_ongpu(l.outputs*l.batch, l.r_gpu, 1, l.forgot_state_gpu, 1); in backward_gru_layer_gpu()
362 …if(prev_delta_gpu) mult_add_into_gpu(l.outputs*l.batch, l.forgot_delta_gpu, l.r_gpu, prev_delta_gp… in backward_gru_layer_gpu()
[all …]
|
| H A D | layer.c | 146 if (l.r_gpu) cuda_free(l.r_gpu); in free_layer_custom()
|
| /dports/misc/darknet/darknet-darknet_yolo_v4_pre/include/ |
| H A D | darknet.h | 518 float *r_gpu; member
|