blame | history | patch | commit | commitdiff | ignore whitespace
Joseph Redmon
2015-05-06 e7688a05a194e3c8baf3c11fbf09b7f5e8640a77 
 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