Searched refs:binary_weights_gpu (Results 1 – 4 of 4) sorted by relevance
| /dports/misc/darknet/darknet-darknet_yolo_v4_pre/src/ |
| H A D | layer.c | 160 if (l.binary_weights_gpu) cuda_free(l.binary_weights_gpu); in free_layer_custom()
|
| H A D | convolutional_layer.c | 35 l->weights_gpu = l->binary_weights_gpu; in swap_binary()
36 l->binary_weights_gpu = swap; in swap_binary()
587 l.binary_weights_gpu = cuda_make_array(l.weights, l.nweights); in make_convolutional_layer()
590 l.binary_weights_gpu = cuda_make_array(l.weights, l.nweights); in make_convolutional_layer()
1040 …status = cudaMemcpy(l->binary_weights_gpu, l->binary_weights, m*k * sizeof(float), cudaMemcpyHostT… in binary_align_weights()
|
| H A D | convolutional_kernels.cu | 170 … binarize_weights_gpu(l.weights_gpu, l.n, (l.c / l.groups)*l.size*l.size, l.binary_weights_gpu); in forward_convolutional_layer_gpu()
178 …_weights_gpu(l.weights_gpu, l.n, (l.c / l.groups)*l.size*l.size, l.binary_weights_gpu, l.mean_arr_… in forward_convolutional_layer_gpu()
|
| /dports/misc/darknet/darknet-darknet_yolo_v4_pre/include/ |
| H A D | darknet.h | 564 float *binary_weights_gpu; member
|