src/layer.h
@@ -28,6 +28,7 @@ CRNN, BATCHNORM, NETWORK, XNOR, BLANK } LAYER_TYPE; @@ -50,6 +51,7 @@ int h,w,c; int out_h, out_w, out_c; int n; int max_boxes; int groups; int size; int side; @@ -101,6 +103,9 @@ char *cfilters; float *filter_updates; float *state; float *prev_state; float *forgot_state; float *forgot_delta; float *state_delta; float *concat; @@ -158,6 +163,12 @@ struct layer *input_h_layer; struct layer *state_h_layer; float *z_cpu; float *r_cpu; float *h_cpu; float *binary_input; size_t workspace_size; #ifdef GPU