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

TensorRT version

Browse Source
This commit is contained in:
Francesco Gatti
2017-08-01 17:51:49 +02:00
parent 1cfe70365f
commit ed5e5d58b5
6 changed files with 16 additions and 11 deletions
Download Patch File Download Diff File Expand all files Collapse all files
src/Reorg.cpp
+1 -1
View File
@@ -26,7 +26,7 @@ Reorg::~Reorg() {
value_type* Reorg::infer(dataDim_t &dim, value_type* srcData) {
reorgForward(srcData, dstData, dim, stride);
reorgForward(srcData, dstData, dim.n, dim.c, dim.h, dim.w, stride);
dim = output_dim;
return dstData;