src/layer.h
@@ -13,7 +13,8 @@ DROPOUT, CROP, ROUTE, COST COST, NORMALIZATION } LAYER_TYPE; typedef enum{ @@ -48,6 +49,10 @@ int does_cost; int joint; float alpha; float beta; float kappa; int dontload; float probability; @@ -69,6 +74,8 @@ int * input_sizes; float * delta; float * output; float * squared; float * norms; #ifdef GPU int *indexes_gpu; @@ -86,6 +93,8 @@ float * output_gpu; float * delta_gpu; float * rand_gpu; float * squared_gpu; float * norms_gpu; #endif } layer;