mirror/darknet
1
0
Fork 0
mirror of https://github.com/pjreddie/darknet.git synced 2023-08-10 21:13:14 +03:00
Code Issues Projects Releases Wiki Activity

multilayer [shortcut] in progress

Browse Source
This commit is contained in:
AlexeyAB
2020-01-05 14:55:21 +03:00
parent 8bd7dcbd58
commit b267e34487
13 changed files with 277 additions and 42 deletions
Download Patch File Download Diff File Expand all files Collapse all files

6
include/darknet.h
View File

@ -328,6 +328,8 @@ struct layer {
int * indexes;
int * input_layers;
int * input_sizes;
float **layers_output;
float **layers_delta;
int * map;
int * counts;
float ** sums;
@ -575,6 +577,10 @@ struct layer {
float *gt_gpu;
float *a_avg_gpu;
int *input_sizes_gpu;
float **layers_output_gpu;
float **layers_delta_gpu;
#ifdef CUDNN
cudnnTensorDescriptor_t srcTensorDesc, dstTensorDesc;
cudnnTensorDescriptor_t srcTensorDesc16, dstTensorDesc16;