Home
last modified time | relevance | path

Searched refs:input_sizes_gpu (Results 1 – 3 of 3) sorted by relevance

/dports/misc/darknet/darknet-darknet_yolo_v4_pre/src/
H A Dshortcut_layer.c83 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 Dlayer.c200 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 Ddarknet.h614 int *input_sizes_gpu; member