From d6cb0fcabcece079c6a5b511159127f585897dba Mon Sep 17 00:00:00 2001 From: Tino Hager <tino.hager@nager.at> Date: Wed, 27 Jun 2018 21:11:33 +0000 Subject: [PATCH] optimize max object definition --- src/maxpool_layer.h | 6 +++--- 1 files changed, 3 insertions(+), 3 deletions(-) diff --git a/src/maxpool_layer.h b/src/maxpool_layer.h index 4456863..ce56dd8 100644 --- a/src/maxpool_layer.h +++ b/src/maxpool_layer.h @@ -2,15 +2,15 @@ #define MAXPOOL_LAYER_H #include "image.h" -#include "params.h" #include "cuda.h" #include "layer.h" +#include "network.h" typedef layer maxpool_layer; image get_maxpool_image(maxpool_layer l); -maxpool_layer make_maxpool_layer(int batch, int h, int w, int c, int size, int stride); -void resize_maxpool_layer(maxpool_layer *l, int h, int w); +maxpool_layer make_maxpool_layer(int batch, int h, int w, int c, int size, int stride, int padding); +void resize_maxpool_layer(maxpool_layer *l, int w, int h); void forward_maxpool_layer(const maxpool_layer l, network_state state); void backward_maxpool_layer(const maxpool_layer l, network_state state); -- Gitblit v1.10.0