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

fixing reorg layer bug

Browse Source
This commit is contained in:
MohammadReza Hosseini
2020-06-21 14:06:12 +04:30
parent c7ceff73ea
commit c49bd1865a
2 changed files with 4 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files
include/tkDNN/pluginsRT/ReorgRT.h
+3 -3
View File
@@ -21,9 +21,9 @@ public:
} }
void configure(const Dims* inputDims, int nbInputs, const Dims* outputDims, int nbOutputs, int maxBatchSize) override { void configure(const Dims* inputDims, int nbInputs, const Dims* outputDims, int nbOutputs, int maxBatchSize) override {
c = inputDims[0].d[0]; c = outputDims[0].d[0];
h = inputDims[0].d[1]; h = outputDims[0].d[1];
w = inputDims[0].d[2]; w = outputDims[0].d[2];
} }
int initialize() override { int initialize() override {
src/Reorg.cpp
+1 -1
View File
@@ -25,7 +25,7 @@ Reorg::~Reorg() {
dnnType* Reorg::infer(dataDim_t &dim, dnnType* srcData) { dnnType* Reorg::infer(dataDim_t &dim, dnnType* srcData) {
reorgForward(srcData, dstData, dim.n, dim.c, dim.h, dim.w, stride); reorgForward(srcData, dstData, output_dim.n, output_dim.c, output_dim.h, output_dim.w, stride);
dim = output_dim; dim = output_dim;
return dstData; return dstData;