blame | history | patch | commit | commitdiff | ignore whitespace
Joseph Redmon
2015-04-15 47528e37cf29e0f9da6885213e5aee044bed84ef 
 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