|
|
@ -22,7 +22,7 @@ void statistics_connected_layer(layer l); |
|
|
|
#ifdef GPU |
|
|
|
#ifdef GPU |
|
|
|
void forward_connected_layer_gpu(connected_layer layer, network_state state); |
|
|
|
void forward_connected_layer_gpu(connected_layer layer, network_state state); |
|
|
|
void backward_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, int batch, float learning_rate, float momentum, float decay); |
|
|
|
void update_connected_layer_gpu(connected_layer layer, int batch, float learning_rate, float momentum, float decay, float loss_scale); |
|
|
|
void push_connected_layer(connected_layer layer); |
|
|
|
void push_connected_layer(connected_layer layer); |
|
|
|
void pull_connected_layer(connected_layer layer); |
|
|
|
void pull_connected_layer(connected_layer layer); |
|
|
|
#endif |
|
|
|
#endif |
|
|
|