TensorRT  6.0.1.5
NvOnnxConfig.h
Go to the documentation of this file.
1 /*
2  * Copyright 1993-2019 NVIDIA Corporation. All rights reserved.
3  *
4  * NOTICE TO LICENSEE:
5  *
6  * This source code and/or documentation ("Licensed Deliverables") are
7  * subject to NVIDIA intellectual property rights under U.S. and
8  * international Copyright laws.
9  *
10  * These Licensed Deliverables contained herein is PROPRIETARY and
11  * CONFIDENTIAL to NVIDIA and is being provided under the terms and
12  * conditions of a form of NVIDIA software license agreement by and
13  * between NVIDIA and Licensee ("License Agreement") or electronically
14  * accepted by Licensee. Notwithstanding any terms or conditions to
15  * the contrary in the License Agreement, reproduction or disclosure
16  * of the Licensed Deliverables to any third party without the express
17  * written consent of NVIDIA is prohibited.
18  *
19  * NOTWITHSTANDING ANY TERMS OR CONDITIONS TO THE CONTRARY IN THE
20  * LICENSE AGREEMENT, NVIDIA MAKES NO REPRESENTATION ABOUT THE
21  * SUITABILITY OF THESE LICENSED DELIVERABLES FOR ANY PURPOSE. IT IS
22  * PROVIDED "AS IS" WITHOUT EXPRESS OR IMPLIED WARRANTY OF ANY KIND.
23  * NVIDIA DISCLAIMS ALL WARRANTIES WITH REGARD TO THESE LICENSED
24  * DELIVERABLES, INCLUDING ALL IMPLIED WARRANTIES OF MERCHANTABILITY,
25  * NONINFRINGEMENT, AND FITNESS FOR A PARTICULAR PURPOSE.
26  * NOTWITHSTANDING ANY TERMS OR CONDITIONS TO THE CONTRARY IN THE
27  * LICENSE AGREEMENT, IN NO EVENT SHALL NVIDIA BE LIABLE FOR ANY
28  * SPECIAL, INDIRECT, INCIDENTAL, OR CONSEQUENTIAL DAMAGES, OR ANY
29  * DAMAGES WHATSOEVER RESULTING FROM LOSS OF USE, DATA OR PROFITS,
30  * WHETHER IN AN ACTION OF CONTRACT, NEGLIGENCE OR OTHER TORTIOUS
31  * ACTION, ARISING OUT OF OR IN CONNECTION WITH THE USE OR PERFORMANCE
32  * OF THESE LICENSED DELIVERABLES.
33  *
34  * U.S. Government End Users. These Licensed Deliverables are a
35  * "commercial item" as that term is defined at 48 C.F.R. 2.101 (OCT
36  * 1995), consisting of "commercial computer software" and "commercial
37  * computer software documentation" as such terms are used in 48
38  * C.F.R. 12.212 (SEPT 1995) and is provided to the U.S. Government
39  * only as a commercial end item. Consistent with 48 C.F.R.12.212 and
40  * 48 C.F.R. 227.7202-1 through 227.7202-4 (JUNE 1995), all
41  * U.S. Government End Users acquire the Licensed Deliverables with
42  * only those rights set forth herein.
43  *
44  * Any use of the Licensed Deliverables in individual and commercial
45  * software must include, in the user documentation and internal
46  * comments to the code, the above Disclaimer and U.S. Government End
47  * Users Notice.
48  */
49 
50 #ifndef NV_OnnxConfig_H
51 #define NV_OnnxConfig_H
52 
53 #include "NvInfer.h"
54 
55 namespace nvonnxparser
56 {
57 
67 
73 
79 {
80 protected:
81  virtual ~IOnnxConfig() {}
82 
83 public:
88  typedef int Verbosity;
89 
97  virtual void setModelDtype(const nvinfer1::DataType) TRTNOEXCEPT = 0;
98 
106  virtual nvinfer1::DataType getModelDtype() const TRTNOEXCEPT = 0;
107 
115  virtual const char* getModelFileName() const TRTNOEXCEPT = 0;
116 
128  virtual void setModelFileName(const char* onnxFilename) TRTNOEXCEPT = 0;
129 
137  virtual Verbosity getVerbosityLevel() const TRTNOEXCEPT = 0;
138 
146  virtual void addVerbosity() TRTNOEXCEPT = 0;
147  virtual void reduceVerbosity() TRTNOEXCEPT = 0;
148  virtual void setVerbosityLevel(Verbosity) TRTNOEXCEPT = 0;
149 
157  virtual const char* getTextFileName() const TRTNOEXCEPT = 0;
158 
170  virtual void setTextFileName(const char* textFileName) TRTNOEXCEPT = 0;
171 
179  virtual const char* getFullTextFileName() const TRTNOEXCEPT = 0;
180 
192  virtual void setFullTextFileName(const char* fullTextFileName) TRTNOEXCEPT = 0;
193 
201  virtual bool getPrintLayerInfo() const TRTNOEXCEPT = 0;
202 
208  virtual void setPrintLayerInfo(bool) TRTNOEXCEPT = 0;
209 
213  virtual void destroy() TRTNOEXCEPT = 0;
214 
215 }; // class IOnnxConfig
216 
217 TENSORRTAPI IOnnxConfig* createONNXConfig();
218 
219 } // namespace nvonnxparser
220 
221 #endif
DataType
The type of weights and tensors.
Definition: NvInferRuntimeCommon.h:162
virtual nvinfer1::DataType getModelDtype() const =0
Get the Model Data Type.
virtual void setTextFileName(const char *textFileName)=0
Set the File Name of the Network Description as a Text File.
virtual void setPrintLayerInfo(bool)=0
Set whether the layer information will be printed.
virtual Verbosity getVerbosityLevel() const =0
Get the Verbosity Level.
virtual void setVerbosityLevel(Verbosity)=0
Set to specific verbosity Level.
virtual void setModelDtype(const nvinfer1::DataType)=0
Set the Model Data Type.
virtual const char * getFullTextFileName() const =0
Get the File Name of the Network Description as a Text File, including the weights.
virtual void addVerbosity()=0
Increase the Verbosity Level.
virtual void destroy()=0
Destroy IOnnxConfig object.
virtual bool getPrintLayerInfo() const =0
Get whether the layer information will be printed.
Definition: NvOnnxConfig.h:55
virtual void setModelFileName(const char *onnxFilename)=0
Set the Model File Name.
int Verbosity
Defines Verbosity level.
Definition: NvOnnxConfig.h:88
virtual void setFullTextFileName(const char *fullTextFileName)=0
Set the File Name of the Network Description as a Text File, including the weights.
virtual void reduceVerbosity()=0
Decrease verbosity Level.
Configuration Manager Class.
Definition: NvOnnxConfig.h:78
virtual const char * getModelFileName() const =0
Get the Model FileName.
virtual const char * getTextFileName() const =0
Returns the File Name of the Network Description as a Text File.