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

stream in TRT plugin

Browse Source
This commit is contained in:
Francesco Gatti
2017-08-10 19:21:15 +02:00
parent 9a6058ac4a
commit 3124f86878
9 changed files with 25 additions and 20 deletions
Download Patch File Download Diff File Expand all files Collapse all files
src/pluginsRT/ReorgRT.cpp
+1 -1
View File
@@ -42,7 +42,7 @@ public:
reorgForward((dnnType*)reinterpret_cast<const dnnType*>(inputs[0]),
reinterpret_cast<dnnType*>(outputs[0]),
batchSize, c, h, w, stride);
batchSize, c, h, w, stride, stream);
return 0;
}