Searched refs:transposed_align_workspace_gpu (Results 1 – 4 of 4) sorted by relevance
/dports/misc/darknet/darknet-darknet_yolo_v4_pre/src/ |
H A D | layer.c | 178 if (l.transposed_align_workspace_gpu) cuda_free(l.transposed_align_workspace_gpu); in free_layer_custom()
|
H A D | convolutional_kernels.cu | 288 …nspose_uint32_gpu((uint32_t *)state.workspace, (uint32_t *)l.transposed_align_workspace_gpu, new_k… in forward_convolutional_layer_gpu() 297 …nsigned char *)l.align_bit_weights_gpu, new_ldb, (unsigned char *)l.transposed_align_workspace_gpu, in forward_convolutional_layer_gpu() 353 …bin_gpu((unsigned char *)state.workspace, (unsigned char *)l.transposed_align_workspace_gpu, k, n,… in forward_convolutional_layer_gpu() 368 …nsigned char *)l.align_bit_weights_gpu, new_ldb, (unsigned char *)l.transposed_align_workspace_gpu, in forward_convolutional_layer_gpu()
|
H A D | convolutional_layer.c | 1032 …status = cudaMalloc((void **)&l->transposed_align_workspace_gpu, l->align_workspace_size * sizeof(… in binary_align_weights()
|
/dports/misc/darknet/darknet-darknet_yolo_v4_pre/include/ |
H A D | darknet.h | 400 float *transposed_align_workspace_gpu; member
|