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

minor migrations

Browse Source
This commit is contained in:
Harshvardhan Chandirasekar
2021-03-17 23:06:50 +05:30
parent f52faeefe4
commit a1c800632d
2 changed files with 8 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files
include/tkDNN/NetworkRT.h
+4 -2
View File
@@ -54,8 +54,10 @@ class NetworkRT {
public:
nvinfer1::DataType dtRT;
nvinfer1::IBuilder *builderRT;
nvinfer1::IRuntime *runtimeRT;
//nvinfer1::IBuilder *builderRT;
std::unique_ptr<nvinfer1::IBuilder,InferDeleter> builderRT;
//nvinfer1::IRuntime *runtimeRT;
std::unique_ptr<nvinfer1::IRuntime,InferDeleter> runtimeRT;
nvinfer1::INetworkDefinition *networkRT;
#if NV_TENSORRT_MAJOR >= 6
nvinfer1::IBuilderConfig *configRT;