maxpool_layer.h 641 B

1234567891011121314151617181920212223
  1. #ifndef MAXPOOL_LAYER_H
  2. #define MAXPOOL_LAYER_H
  3. #include "image.h"
  4. #include "cuda.h"
  5. #include "layer.h"
  6. #include "network.h"
  7. typedef layer maxpool_layer;
  8. image get_maxpool_image(maxpool_layer l);
  9. maxpool_layer make_maxpool_layer(int batch, int h, int w, int c, int size, int stride, int padding);
  10. void resize_maxpool_layer(maxpool_layer *l, int w, int h);
  11. void forward_maxpool_layer(const maxpool_layer l, network net);
  12. void backward_maxpool_layer(const maxpool_layer l, network net);
  13. #ifdef GPU
  14. void forward_maxpool_layer_gpu(maxpool_layer l, network net);
  15. void backward_maxpool_layer_gpu(maxpool_layer l, network net);
  16. #endif
  17. #endif