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

fix

Browse Source
This commit is contained in:
fbagni
2019-10-30 09:42:54 +01:00
parent 33f36ab204
commit d6d93a74f8
1 changed files with 2 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files
src/NetworkRT.cpp
+2 -2
View File
@@ -236,14 +236,14 @@ ILayer* NetworkRT::convert_layer(ITensor *input, Conv2d *l) {
if(!l->deConv) { if(!l->deConv) {
IConvolutionLayer *lRTconv = networkRT->addConvolution(*input, IConvolutionLayer *lRTconv = networkRT->addConvolution(*input,
l->outputs, DimsHW{l->kernelH, l->kernelW}, w, b); l->outputs, DimsHW{l->kernelH, l->kernelW}, w, b);
checkNULL(lRT); 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->paddingW});
lRT = (ILayer*) lRTconv; lRT = (ILayer*) lRTconv;
} else { } else {
IDeconvolutionLayer *lRTconv = networkRT->addDeconvolution(*input, IDeconvolutionLayer *lRTconv = networkRT->addDeconvolution(*input,
l->outputs, DimsHW{l->kernelH, l->kernelW}, w, b); l->outputs, DimsHW{l->kernelH, l->kernelW}, w, b);
checkNULL(lRT); 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->paddingW});
lRT = (ILayer*) lRTconv; lRT = (ILayer*) lRTconv;