Go to the documentation of this file.00001 #ifndef LOCAL_LAYER_H
00002 #define LOCAL_LAYER_H
00003
00004 #include "cuda.h"
00005 #include "image.h"
00006 #include "activations.h"
00007 #include "layer.h"
00008 #include "network.h"
00009
00010 typedef layer local_layer;
00011
00012 #ifdef GPU
00013 void forward_local_layer_gpu(local_layer layer, network_state state);
00014 void backward_local_layer_gpu(local_layer layer, network_state state);
00015 void update_local_layer_gpu(local_layer layer, int batch, float learning_rate, float momentum, float decay);
00016
00017 void push_local_layer(local_layer layer);
00018 void pull_local_layer(local_layer layer);
00019 #endif
00020
00021 local_layer make_local_layer(int batch, int h, int w, int c, int n, int size, int stride, int pad, ACTIVATION activation);
00022
00023 void forward_local_layer(const local_layer layer, network_state state);
00024 void backward_local_layer(local_layer layer, network_state state);
00025 void update_local_layer(local_layer layer, int batch, float learning_rate, float momentum, float decay);
00026
00027 void bias_output(float *output, float *biases, int batch, int n, int size);
00028 void backward_bias(float *bias_updates, float *delta, int batch, int n, int size);
00029
00030 #endif
00031