| | |
| | | #define NORMALIZATION_LAYER_H |
| | | |
| | | #include "image.h" |
| | | #include "layer.h" |
| | | #include "network.h" |
| | | |
| | | typedef struct { |
| | | int batch; |
| | | int h,w,c; |
| | | int size; |
| | | float alpha; |
| | | float beta; |
| | | float kappa; |
| | | float *delta; |
| | | float *output; |
| | | float *sums; |
| | | } normalization_layer; |
| | | layer make_normalization_layer(int batch, int w, int h, int c, int size, float alpha, float beta, float kappa); |
| | | void resize_normalization_layer(layer *layer, int h, int w); |
| | | void forward_normalization_layer(const layer layer, network_state state); |
| | | void backward_normalization_layer(const layer layer, network_state state); |
| | | void visualize_normalization_layer(layer layer, char *window); |
| | | |
| | | image get_normalization_image(normalization_layer layer); |
| | | normalization_layer *make_normalization_layer(int batch, int h, int w, int c, int size, float alpha, float beta, float kappa); |
| | | void resize_normalization_layer(normalization_layer *layer, int h, int w, int c); |
| | | void forward_normalization_layer(const normalization_layer layer, float *in); |
| | | void backward_normalization_layer(const normalization_layer layer, float *in, float *delta); |
| | | void visualize_normalization_layer(normalization_layer layer, char *window); |
| | | |
| | | #ifdef GPU |
| | | void forward_normalization_layer_gpu(const layer layer, network_state state); |
| | | void backward_normalization_layer_gpu(const layer layer, network_state state); |
| | | #endif |
| | | |
| | | #endif |