mirror of
https://github.com/pjreddie/darknet.git
synced 2023-08-10 21:13:14 +03:00
so much need to commit
This commit is contained in:
17
src/batchnorm_layer.h
Normal file
17
src/batchnorm_layer.h
Normal file
@@ -0,0 +1,17 @@
|
||||
#ifndef BATCHNORM_LAYER_H
|
||||
#define BATCHNORM_LAYER_H
|
||||
|
||||
#include "image.h"
|
||||
#include "layer.h"
|
||||
#include "network.h"
|
||||
|
||||
layer make_batchnorm_layer(int batch, int w, int h, int c);
|
||||
void forward_batchnorm_layer(layer l, network_state state);
|
||||
void backward_batchnorm_layer(layer l, network_state state);
|
||||
|
||||
#ifdef GPU
|
||||
void forward_batchnorm_layer_gpu(layer l, network_state state);
|
||||
void backward_batchnorm_layer_gpu(layer l, network_state state);
|
||||
#endif
|
||||
|
||||
#endif
|
||||
Reference in New Issue
Block a user