AlexeyAB
2018-08-23 5b6be00d4b1ffd671c20c4c72d2239c924eaa3d4
src/layer.h
@@ -78,7 +78,6 @@
    int out_h, out_w, out_c;
    int n;
    int max_boxes;
   int small_object;
    int groups;
    int size;
    int side;
@@ -114,6 +113,7 @@
   int tanh;
   int *mask;
   int total;
   float bflops;
    int adam;
    float B1;
@@ -179,6 +179,11 @@
    float *weights;
    float *weight_updates;
    char *align_bit_weights;
    float *mean_arr;
    int lda_align;
    int bit_align;
    float *col_image;
    int   * input_layers;
    int   * input_sizes;
@@ -282,7 +287,7 @@
    #ifdef CUDNN
    cudnnTensorDescriptor_t srcTensorDesc, dstTensorDesc;
    cudnnTensorDescriptor_t dsrcTensorDesc, ddstTensorDesc;
   cudnnTensorDescriptor_t normTensorDesc;
   cudnnTensorDescriptor_t normTensorDesc, normDstTensorDesc, normDstTensorDescF16;
    cudnnFilterDescriptor_t weightDesc;
    cudnnFilterDescriptor_t dweightDesc;
    cudnnConvolutionDescriptor_t convDesc;