From 168af40fe9a3cc81c6ee16b3e81f154780c36bdb Mon Sep 17 00:00:00 2001 From: Scheaven <xuepengqiang> Date: 星期四, 03 六月 2021 15:03:27 +0800 Subject: [PATCH] up new v4 --- lib/detecter_tools/darknet/maxpool_layer.h | 72 ++++++++++++++++++------------------ 1 files changed, 36 insertions(+), 36 deletions(-) diff --git a/lib/detecter_tools/darknet/maxpool_layer.h b/lib/detecter_tools/darknet/maxpool_layer.h index 967806a..3c00611 100644 --- a/lib/detecter_tools/darknet/maxpool_layer.h +++ b/lib/detecter_tools/darknet/maxpool_layer.h @@ -1,36 +1,36 @@ -#ifndef MAXPOOL_LAYER_H -#define MAXPOOL_LAYER_H - -#include "image.h" -#include "dark_cuda.h" -#include "layer.h" -#include "network.h" - -typedef layer maxpool_layer; - -#ifdef __cplusplus -extern "C" { -#endif -image get_maxpool_image(maxpool_layer l); -maxpool_layer make_maxpool_layer(int batch, int h, int w, int c, int size, int stride_x, int stride_y, int padding, int maxpool_depth, int out_channels, int antialiasing, int avgpool, int train); -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); - -void forward_local_avgpool_layer(const maxpool_layer l, network_state state); -void backward_local_avgpool_layer(const maxpool_layer l, network_state state); - -#ifdef GPU -void forward_maxpool_layer_gpu(maxpool_layer l, network_state state); -void backward_maxpool_layer_gpu(maxpool_layer l, network_state state); -void cudnn_maxpool_setup(maxpool_layer *l); - -void forward_local_avgpool_layer_gpu(maxpool_layer layer, network_state state); -void backward_local_avgpool_layer_gpu(maxpool_layer layer, network_state state); -#endif // GPU - -#ifdef __cplusplus -} -#endif - -#endif +#ifndef MAXPOOL_LAYER_H +#define MAXPOOL_LAYER_H + +#include "image.h" +#include "dark_cuda.h" +#include "layer.h" +#include "network.h" + +typedef layer maxpool_layer; + +#ifdef __cplusplus +extern "C" { +#endif +image get_maxpool_image(maxpool_layer l); +maxpool_layer make_maxpool_layer(int batch, int h, int w, int c, int size, int stride_x, int stride_y, int padding, int maxpool_depth, int out_channels, int antialiasing, int avgpool, int train); +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); + +void forward_local_avgpool_layer(const maxpool_layer l, network_state state); +void backward_local_avgpool_layer(const maxpool_layer l, network_state state); + +#ifdef GPU +void forward_maxpool_layer_gpu(maxpool_layer l, network_state state); +void backward_maxpool_layer_gpu(maxpool_layer l, network_state state); +void cudnn_maxpool_setup(maxpool_layer *l); + +void forward_local_avgpool_layer_gpu(maxpool_layer layer, network_state state); +void backward_local_avgpool_layer_gpu(maxpool_layer layer, network_state state); +#endif // GPU + +#ifdef __cplusplus +} +#endif + +#endif -- Gitblit v1.8.0