TensorRT 8.5.3
NvOnnxConfig.h
Go to the documentation of this file.
1/*
2 * SPDX-FileCopyrightText: Copyright (c) 1993-2022 NVIDIA CORPORATION & AFFILIATES. All rights reserved.
3 * SPDX-License-Identifier: LicenseRef-NvidiaProprietary
4 *
5 * NVIDIA CORPORATION, its affiliates and licensors retain all intellectual
6 * property and proprietary rights in and to this material, related
7 * documentation and any modifications thereto. Any use, reproduction,
8 * disclosure or distribution of this material and related documentation
9 * without an express license agreement from NVIDIA CORPORATION or
10 * its affiliates is strictly prohibited.
11 */
12
13#ifndef NV_OnnxConfig_H
14#define NV_OnnxConfig_H
15
16#include "NvInfer.h"
17
18namespace nvonnxparser
19{
20
30
36
42{
43public:
44 virtual ~IOnnxConfig() noexcept = default;
49 typedef int32_t Verbosity;
50
59 virtual void setModelDtype(const nvinfer1::DataType) noexcept = 0;
60
68 virtual nvinfer1::DataType getModelDtype() const noexcept = 0;
69
77 virtual char const* getModelFileName() const noexcept = 0;
78
90 virtual void setModelFileName(char const* onnxFilename) noexcept = 0;
91
99 virtual Verbosity getVerbosityLevel() const noexcept = 0;
100
108 virtual void addVerbosity() noexcept = 0;
109
115 virtual void reduceVerbosity() noexcept = 0;
116
122 virtual void setVerbosityLevel(Verbosity) noexcept = 0;
123
132 virtual char const* getTextFileName() const noexcept = 0;
133
145 virtual void setTextFileName(char const* textFileName) noexcept = 0;
146
155 virtual char const* getFullTextFileName() const noexcept = 0;
156
168 virtual void setFullTextFileName(char const* fullTextFileName) noexcept = 0;
169
177 virtual bool getPrintLayerInfo() const noexcept = 0;
178
184 virtual void setPrintLayerInfo(bool) noexcept = 0;
185
193 TRT_DEPRECATED virtual void destroy() noexcept = 0;
194
195}; // class IOnnxConfig
196
198
199} // namespace nvonnxparser
200
201#endif
#define TENSORRTAPI
Definition: NvInferRuntimeCommon.h:54
#define TRT_DEPRECATED
Definition: NvInferRuntimeCommon.h:40
Configuration Manager Class.
Definition: NvOnnxConfig.h:42
virtual void setFullTextFileName(char const *fullTextFileName) noexcept=0
Set the File Name of the Network Description as a Text File, including the weights.
virtual char const * getTextFileName() const noexcept=0
Returns the File Name of the Network Description as a Text File.
virtual void addVerbosity() noexcept=0
Increase the Verbosity Level.
virtual nvinfer1::DataType getModelDtype() const noexcept=0
Get the Model Data Type.
virtual void setPrintLayerInfo(bool) noexcept=0
Set whether the layer information will be printed.
virtual void reduceVerbosity() noexcept=0
Reduce the Verbosity Level.
int32_t Verbosity
Defines Verbosity level.
Definition: NvOnnxConfig.h:49
virtual Verbosity getVerbosityLevel() const noexcept=0
Get the Verbosity Level.
virtual bool getPrintLayerInfo() const noexcept=0
Get whether the layer information will be printed.
virtual void setModelDtype(const nvinfer1::DataType) noexcept=0
Set the Model Data Type.
virtual void setVerbosityLevel(Verbosity) noexcept=0
Set to specific verbosity Level.
virtual ~IOnnxConfig() noexcept=default
virtual void setModelFileName(char const *onnxFilename) noexcept=0
Set the Model File Name.
virtual char const * getModelFileName() const noexcept=0
Get the Model FileName.
virtual char const * getFullTextFileName() const noexcept=0
Get the File Name of the Network Description as a Text File, including the weights.
virtual void setTextFileName(char const *textFileName) noexcept=0
Set the File Name of the Network Description as a Text File.
virtual TRT_DEPRECATED void destroy() noexcept=0
Destroy IOnnxConfig object.
The TensorRT API version 1 namespace.
DataType
The type of weights and tensors.
Definition: NvInferRuntimeCommon.h:117
The TensorRT ONNX parser API namespace.
Definition: NvOnnxConfig.h:19
IOnnxConfig * createONNXConfig()

  Copyright © 2024 NVIDIA Corporation
  Privacy Policy | Manage My Privacy | Do Not Sell or Share My Data | Terms of Service | Accessibility | Corporate Policies | Product Security | Contact