Searched refs:input_sizes_gpu (Results 1 – 3 of 3) sorted by relevance
/dports/misc/darknet/darknet-darknet_yolo_v4_pre/src/ |
H A D | shortcut_layer.c | 83 l.input_sizes_gpu = cuda_make_int_array_new_api(input_sizes, l.n); in make_shortcut_layer() 135 memcpy_ongpu(l->input_sizes_gpu, l->input_sizes, l->n * sizeof(int)); in resize_shortcut_layer() 221 …shortcut_multilayer_gpu(l.outputs, l.batch, l.n, l.input_sizes_gpu, l.layers_output_gpu, l.output_… in forward_shortcut_layer_gpu() 236 …backward_shortcut_multilayer_gpu(l.outputs, l.batch, l.n, l.input_sizes_gpu, l.layers_delta_gpu, s… in backward_shortcut_layer_gpu()
|
H A D | layer.c | 200 if (l.input_sizes_gpu) cuda_free((float*)l.input_sizes_gpu); in free_layer_custom()
|
/dports/misc/darknet/darknet-darknet_yolo_v4_pre/include/ |
H A D | darknet.h | 614 int *input_sizes_gpu; member
|