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

2 input shortcut

Browse Source
This commit is contained in:
Francesco Gatti
2018-12-21 16:16:38 +01:00
parent 64626bf547
commit 53b429551d
2 changed files with 9 additions and 6 deletions
Download Patch File Download Diff File Expand all files Collapse all files
src/NetworkRT.cpp
+5 -3
View File
@@ -328,9 +328,11 @@ ILayer* NetworkRT::convert_layer(ITensor *input, Shortcut *l) {
//std::cout<<"convert Shortcut\n";
//std::cout<<"New plugin Shortcut\n";
ITensor *tens = tensors[l->backLayer];
IPlugin *plugin = new ShortcutRT(tens);
IPluginLayer *lRT = networkRT->addPlugin(&input, 1, *plugin);
ITensor *back_tens = tensors[l->backLayer];
IPlugin *plugin = new ShortcutRT();
ITensor *inputs[2] = { input, back_tens };
IPluginLayer *lRT = networkRT->addPlugin(inputs, 2, *plugin);
checkNULL(lRT);
return lRT;
}