blame | history | patch | commit | commitdiff | ignore whitespace
Joseph Redmon
2015-04-11 390a0cf923cee683e5be300390c736a2ab9b7fd5 
 src/connected_layer.h


@@ -38,12 +38,12 @@




void forward_connected_layer(connected_layer layer, network_state state);


void backward_connected_layer(connected_layer layer, network_state state);


void update_connected_layer(connected_layer layer, float learning_rate, float momentum, float decay);


void update_connected_layer(connected_layer layer, int batch, float learning_rate, float momentum, float decay);




#ifdef GPU


void forward_connected_layer_gpu(connected_layer layer, network_state state);


void backward_connected_layer_gpu(connected_layer layer, network_state state);


void update_connected_layer_gpu(connected_layer layer, float learning_rate, float momentum, float decay);


void update_connected_layer_gpu(connected_layer layer, int batch, float learning_rate, float momentum, float decay);


void push_connected_layer(connected_layer layer);


void pull_connected_layer(connected_layer layer);


#endif