mirror/darknet
1
0
Fork 0
mirror of https://github.com/pjreddie/darknet.git synced 2023-08-10 21:13:14 +03:00
Code Issues Projects Releases Wiki Activity

im2col slightly faster

Browse Source
This commit is contained in:
Joseph Redmon
2014-10-29 23:26:41 -07:00
parent af4e4f92dc
commit 1c0fd9bb47
9 changed files with 115 additions and 63 deletions
Download Patch File Download Diff File Expand all files Collapse all files

8
src/connected_layer.c
View File

@ -135,9 +135,7 @@ void forward_connected_layer_gpu(connected_layer layer, cl_mem input)
{
int i;
for(i = 0; i < layer.batch; ++i){
cl_mem sub = cl_sub_array(layer.output_cl, i*layer.outputs, layer.outputs);
copy_ongpu(layer.outputs, layer.biases_cl, 1, sub, 1);
clReleaseMemObject(sub);
copy_ongpu_offset(layer.outputs, layer.biases_cl, 0, 1, layer.output_cl, i*layer.outputs, 1);
}
int m = layer.batch;
int k = layer.inputs;
@ -154,9 +152,7 @@ void backward_connected_layer_gpu(connected_layer layer, cl_mem input, cl_mem de
int i;
gradient_array_ongpu(layer.output_cl, layer.outputs*layer.batch, layer.activation, layer.delta_cl);
for(i = 0; i < layer.batch; ++i){
cl_mem sub = cl_sub_array(layer.delta_cl, i*layer.outputs, layer.outputs);
axpy_ongpu(layer.outputs, 1, sub, 1, layer.bias_updates_cl, 1);
clReleaseMemObject(sub);
axpy_ongpu_offset(layer.outputs, 1, layer.delta_cl, i*layer.outputs, 1, layer.bias_updates_cl, 0, 1);
}
int m = layer.inputs;
int k = layer.batch;