-
Notifications
You must be signed in to change notification settings - Fork 543
/
NvOnnxParser.cpp
27 lines (23 loc) · 841 Bytes
/
NvOnnxParser.cpp
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
/*
* SPDX-License-Identifier: Apache-2.0
*/
#include "NvOnnxParser.h"
#include "ModelImporter.hpp"
#include "ModelRefitter.hpp"
#include "NvInferRuntime.h"
extern "C" void* createNvOnnxParser_INTERNAL(void* network_, void* logger_, int version) noexcept
{
auto network = static_cast<nvinfer1::INetworkDefinition*>(network_);
auto logger = static_cast<nvinfer1::ILogger*>(logger_);
return new onnx2trt::ModelImporter(network, logger);
}
extern "C" void* createNvOnnxParserRefitter_INTERNAL(void* refitter_, void* logger_, int32_t version) noexcept
{
auto refitter = static_cast<nvinfer1::IRefitter*>(refitter_);
auto logger = static_cast<nvinfer1::ILogger*>(logger_);
return new onnx2trt::ModelRefitter(refitter, logger);
}
extern "C" int getNvOnnxParserVersion() noexcept
{
return NV_ONNX_PARSER_VERSION;
}