1 #ifndef CONNECTED_LAYER_H 2 #define CONNECTED_LAYER_H 3 4 #include "activations.h" 5 #include "layer.h" 6 #include "network.h" 7 8 typedef layer connected_layer; 9 10 #ifdef __cplusplus 11 extern "C" { 12 #endif 13 connected_layer make_connected_layer(int batch, int steps, int inputs, int outputs, ACTIVATION activation, int batch_normalize); 14 size_t get_connected_workspace_size(layer l); 15 16 void forward_connected_layer(connected_layer layer, network_state state); 17 void backward_connected_layer(connected_layer layer, network_state state); 18 void update_connected_layer(connected_layer layer, int batch, float learning_rate, float momentum, float decay); 19 void denormalize_connected_layer(layer l); 20 void statistics_connected_layer(layer l); 21 22 #ifdef GPU 23 void forward_connected_layer_gpu(connected_layer layer, network_state state); 24 void backward_connected_layer_gpu(connected_layer layer, network_state state); 25 void update_connected_layer_gpu(connected_layer layer, int batch, float learning_rate, float momentum, float decay, float loss_scale); 26 void push_connected_layer(connected_layer layer); 27 void pull_connected_layer(connected_layer layer); 28 #endif 29 30 #ifdef __cplusplus 31 } 32 #endif 33 34 #endif 35