00001 #ifndef ROUTE_LAYER_H 00002 #define ROUTE_LAYER_H 00003 #include "network.h" 00004 #include "layer.h" 00005 00006 typedef layer route_layer; 00007 00008 route_layer make_route_layer(int batch, int n, int *input_layers, int *input_size); 00009 void forward_route_layer(const route_layer l, network_state state); 00010 void backward_route_layer(const route_layer l, network_state state); 00011 void resize_route_layer(route_layer *l, network *net); 00012 00013 #ifdef GPU 00014 void forward_route_layer_gpu(const route_layer l, network_state state); 00015 void backward_route_layer_gpu(const route_layer l, network_state state); 00016 #endif 00017 00018 #endif