00001 #ifndef AVGPOOL_LAYER_H 00002 #define AVGPOOL_LAYER_H 00003 00004 #include "image.h" 00005 #include "cuda.h" 00006 #include "layer.h" 00007 #include "network.h" 00008 00009 typedef layer avgpool_layer; 00010 00011 image get_avgpool_image(avgpool_layer l); 00012 avgpool_layer make_avgpool_layer(int batch, int w, int h, int c); 00013 void resize_avgpool_layer(avgpool_layer *l, int w, int h); 00014 void forward_avgpool_layer(const avgpool_layer l, network_state state); 00015 void backward_avgpool_layer(const avgpool_layer l, network_state state); 00016 00017 #ifdef GPU 00018 void forward_avgpool_layer_gpu(avgpool_layer l, network_state state); 00019 void backward_avgpool_layer_gpu(avgpool_layer l, network_state state); 00020 #endif 00021 00022 #endif 00023