mmr/tkDNN
Archived
1
0
Fork 0
Code Issues 46 Pull Requests 12 Releases 4 Activity

yolo3 ok

Browse Source
This commit is contained in:
Francesco Gatti
2019-01-04 22:28:10 +01:00
parent 2e8d0b1002
commit 88097a3774
7 changed files with 139 additions and 23 deletions
Download Patch File Download Diff File Expand all files Collapse all files
src/pluginsRT/ShortcutRT.cpp
+1 -1
View File
@@ -44,7 +44,7 @@ public:
dnnType *dstData = reinterpret_cast<dnnType*>(outputs[0]);
checkCuda( cudaMemcpyAsync(dstData, srcData, batchSize*c*h*w*sizeof(dnnType), cudaMemcpyDeviceToDevice, stream));
shortcutForward(srcDataBack, dstData, batchSize, c, h, w, 1, batchSize, c, h, w, 1);
shortcutForward(srcDataBack, dstData, batchSize, c, h, w, 1, batchSize, c, h, w, 1, stream);
return 0;
}