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

upsample ok, route have problems

Browse Source
This commit is contained in:
Francesco Gatti
2018-12-22 23:56:01 +01:00
parent 34be4cd00f
commit 3bd725801d
6 changed files with 112 additions and 16 deletions
Download Patch File Download Diff File Expand all files Collapse all files
include/NetworkRT.h
+1
View File
@@ -45,6 +45,7 @@ public:
nvinfer1::ILayer* convert_layer(nvinfer1::ITensor *input, Region *l);
nvinfer1::ILayer* convert_layer(nvinfer1::ITensor *input, Shortcut *l);
nvinfer1::ILayer* convert_layer(nvinfer1::ITensor *input, Yolo *l);
nvinfer1::ILayer* convert_layer(nvinfer1::ITensor *input, Upsample *l);
bool serialize(const char *filename);
bool deserialize(const char *filename);