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

Adding dilated convolution and support for reorg3d layer #47

Open
mrhosseini wants to merge 8 commits from mrhosseini/master into master
pull from: mrhosseini/master
merge into: mmr:master
Conversation 1 Commits 8 Files Changed 8
+2 -3
1 changed files with 2 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files
Showing only changes of commit 91ba13b27a - Show all commits
src/NetworkRT.cpp
+2 -3
View File
@@ -316,9 +316,8 @@ ILayer* NetworkRT::convert_layer(ITensor *input, Conv2d *l) {
l->outputs, DimsHW{l->kernelH, l->kernelW}, w, b); l->outputs, DimsHW{l->kernelH, l->kernelW}, w, b);
checkNULL(lRTconv); checkNULL(lRTconv);
lRTconv->setStride(DimsHW{l->strideH, l->strideW}); lRTconv->setStride(DimsHW{l->strideH, l->strideW});
lRTconv->setPadding(DimsHW{l->paddingH, l->paddingW}); lRTconv->setPadding(DimsHW{l->paddingH * l->dilationH, l->paddingW * l->dilationW});
//std::cout << "dilation: " << l->dilationH << ", " << l->dilationW << std::endl; lRTconv->setDilation(DimsHW{l->dilationH, l->dilationW});
lRTconv->setDilation(DimsHW{l->dilationH, l->dilationW}); //mrho
lRTconv->setNbGroups(l->groups); lRTconv->setNbGroups(l->groups);
lRT = (ILayer*) lRTconv; lRT = (ILayer*) lRTconv;
} else { } else {