blame | history | patch | commit | commitdiff | ignore whitespace
AlexeyAB
2017-10-21 ae74d0ef31485f84e1856b4733135d2753dbb033 
 src/layer.h


@@ -3,6 +3,9 @@




#include "activations.h"


#include "stddef.h"


#include "tree.h"




struct network_state;




struct layer;


typedef struct layer layer;


@@ -42,6 +45,12 @@


    LAYER_TYPE type;


    ACTIVATION activation;


    COST_TYPE cost_type;


    void (*forward)   (struct layer, struct network_state);


    void (*backward)  (struct layer, struct network_state);


    void (*update)    (struct layer, int, float, float, float);


    void (*forward_gpu)   (struct layer, struct network_state);


    void (*backward_gpu)  (struct layer, struct network_state);


    void (*update_gpu)    (struct layer, int, float, float, float);


    int batch_normalize;


    int shortcut;


    int batch;


@@ -58,6 +67,7 @@


    int size;


    int side;


    int stride;


    int reverse;


    int pad;


    int sqrt;


    int flip;


@@ -72,6 +82,7 @@


    float saturation;


    float exposure;


    float shift;


    float ratio;


    int softmax;


    int classes;


    int coords;


@@ -82,6 +93,20 @@


    int joint;


    int noadjust;


    int reorg;


    int log;




    int adam;


    float B1;


    float B2;


    float eps;


    float *m_gpu;


    float *v_gpu;


    int t;


    float *m;


    float *v;




    tree *softmax_tree;


    int  *map;




    float alpha;


    float beta;


@@ -91,7 +116,11 @@


    float object_scale;


    float noobject_scale;


    float class_scale;


    int bias_match;


    int random;


    float thresh;


    int classfix;


    int absolute;




    int dontload;


    int dontloadscales;


@@ -103,9 +132,7 @@


    int *indexes;


    float *rand;


    float *cost;


    float *filters;


    char  *cfilters;


    float *filter_updates;


    char  *cweights;


    float *state;


    float *prev_state;


    float *forgot_state;


@@ -115,7 +142,7 @@


    float *concat;


    float *concat_delta;




    float *binary_filters;


    float *binary_weights;




    float *biases;


    float *bias_updates;


@@ -192,11 +219,9 @@


    float * save_delta_gpu;


    float * concat_gpu;


    float * concat_delta_gpu;


    float * filters_gpu;


    float * filter_updates_gpu;




    float *binary_input_gpu;


    float *binary_filters_gpu;


    float *binary_weights_gpu;




    float * mean_gpu;


    float * variance_gpu;


@@ -228,8 +253,8 @@


    #ifdef CUDNN


    cudnnTensorDescriptor_t srcTensorDesc, dstTensorDesc;


    cudnnTensorDescriptor_t dsrcTensorDesc, ddstTensorDesc;


    cudnnFilterDescriptor_t filterDesc;


    cudnnFilterDescriptor_t dfilterDesc;


    cudnnFilterDescriptor_t weightDesc;


    cudnnFilterDescriptor_t dweightDesc;


    cudnnConvolutionDescriptor_t convDesc;


    cudnnConvolutionFwdAlgo_t fw_algo;


    cudnnConvolutionBwdDataAlgo_t bd_algo;