blame | history | patch | commit | commitdiff | ignore whitespace
Joseph Redmon
2015-03-30 6553b3f0e3e55fc30a99c7d4b5798aa86d18a114 
 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