blame | history | patch | commit | commitdiff | ignore whitespace
Joseph Redmon
2016-05-06 c7b10ceadb1a78e7480d281444a31ae2a7dc1b05 
 src/layer.h


@@ -20,11 +20,18 @@


    NORMALIZATION,


    AVGPOOL,


    LOCAL,


    SHORTCUT


    SHORTCUT,


    ACTIVE,


    RNN,


    GRU,


    CRNN,


    BATCHNORM,


    NETWORK,


    BLANK


} LAYER_TYPE;




typedef enum{


    SSE, MASKED


    SSE, MASKED, SMOOTH


} COST_TYPE;




struct layer{


@@ -32,6 +39,7 @@


    ACTIVATION activation;


    COST_TYPE cost_type;


    int batch_normalize;


    int shortcut;


    int batch;


    int forced;


    int flipped;


@@ -46,11 +54,14 @@


    int side;


    int stride;


    int pad;


    int crop_width;


    int crop_height;


    int sqrt;


    int flip;


    int index;


    int binary;


    int xnor;


    int steps;


    int hidden;


    float dot;


    float angle;


    float jitter;


    float saturation;


@@ -78,6 +89,7 @@


    int dontload;


    int dontloadscales;




    float temperature;


    float probability;


    float scale;




@@ -85,7 +97,15 @@


    float *rand;


    float *cost;


    float *filters;


    char  *cfilters;


    float *filter_updates;


    float *state;


    float *state_delta;




    float *concat;


    float *concat_delta;




    float *binary_filters;




    float *biases;


    float *bias_updates;


@@ -108,16 +128,57 @@


    float * mean;


    float * variance;




    float * mean_delta;


    float * variance_delta;




    float * rolling_mean;


    float * rolling_variance;




    float * x;


    float * x_norm;




    struct layer *input_layer;


    struct layer *self_layer;


    struct layer *output_layer;




    struct layer *input_gate_layer;


    struct layer *state_gate_layer;


    struct layer *input_save_layer;


    struct layer *state_save_layer;


    struct layer *input_state_layer;


    struct layer *state_state_layer;




    struct layer *input_z_layer;


    struct layer *state_z_layer;




    struct layer *input_r_layer;


    struct layer *state_r_layer;




    struct layer *input_h_layer;


    struct layer *state_h_layer;




    #ifdef GPU


    float *z_gpu;


    float *r_gpu;


    float *h_gpu;




    int *indexes_gpu;


    float * prev_state_gpu;


    float * forgot_state_gpu;


    float * forgot_delta_gpu;


    float * state_gpu;


    float * state_delta_gpu;


    float * gate_gpu;


    float * gate_delta_gpu;


    float * save_gpu;


    float * save_delta_gpu;


    float * concat_gpu;


    float * concat_delta_gpu;


    float * filters_gpu;


    float * filter_updates_gpu;




    float * spatial_mean_gpu;


    float * spatial_variance_gpu;


    float *binary_input_gpu;


    float *binary_filters_gpu;




    float * mean_gpu;


    float * variance_gpu;


@@ -125,9 +186,6 @@


    float * rolling_mean_gpu;


    float * rolling_variance_gpu;




    float * spatial_mean_delta_gpu;


    float * spatial_variance_delta_gpu;




    float * variance_delta_gpu;


    float * mean_delta_gpu;