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

Fixed tensorrt8 branch to work jetpack 4.5 and tensorrt7

Browse Source 
Signed-off-by: perseusdg <f20180523@goa.bits-pilani.ac.in>
This commit was merged in pull request #278.
This commit is contained in:
Harshvardhan Chandirasekar
2022-03-16 18:09:32 +05:30
committed by perseusdg
parent 480b5a9c5a
commit 40266a6c32
3 changed files with 17 additions and 7 deletions
Download Patch File Download Diff File Expand all files Collapse all files
include/tkDNN/NetworkRT.h
+1 -1
View File
@@ -95,7 +95,7 @@ public:
nvinfer1::IPluginV2Layer* convert_layer(nvinfer1::ITensor *input, Region *l);
nvinfer1::ILayer* convert_layer(nvinfer1::ITensor *input, Shortcut *l);
nvinfer1::IPluginV2Layer* convert_layer(nvinfer1::ITensor *input, Yolo *l);
nvinfer1::IResizeLayer* convert_layer(nvinfer1::ITensor *input, Upsample *l);
nvinfer1::ILayer* convert_layer(nvinfer1::ITensor *input, Upsample *l);
nvinfer1::ILayer* convert_layer(nvinfer1::ITensor *input, DeformConv2d *l);
nvinfer1::ILayer* convert_layer(nvinfer1::ITensor *input,Padding *l);
nvinfer1::ILayer* convert_layer(nvinfer1::ITensor* input,MulAdd *l);