NvOnnxConfig.h

前往此文件的文档。

/*
 * SPDX-FileCopyrightText: 版权所有 (c) 1993-2024 NVIDIA CORPORATION & AFFILIATES。保留所有权利。
 * SPDX-License-Identifier: Apache-2.0
 *
 * 根据 Apache 许可证 2.0 版本（“许可证”）获得许可；
 * 除非遵守许可证的规定，否则您不得使用此文件。
 * 您可以在以下位置获取许可证副本：
 *
 * https://apache.ac.cn/licenses/LICENSE-2.0
 *
 * 除非适用法律要求或书面同意，根据许可证分发的软件
 * 是在“按现状”基础上分发的，
 * 不附带任何种类的明示或暗示的保证或条件。
 * 有关许可证下特定语言的权限和限制，请参阅许可证。
 * 限制。
 */
 
#ifndef NV_OnnxConfig_H
#define NV_OnnxConfig_H
 
#include "NvInfer.h"
 
namespace nvonnxparser
{
 
 
 
class IOnnxConfig
{
public
 virtual ~IOnnxConfig() noexcept = default;
 typedef int32_t Verbosity;
 
 virtual void setModelDtype(const nvinfer1::DataType) noexcept = 0;
 
 virtual nvinfer1::DataType getModelDtype() const noexcept = 0;
 
 virtual char const* getModelFileName() const noexcept = 0;
 
 virtual void setModelFileName(char const* onnxFilename) noexcept = 0;
 
 virtual Verbosity getVerbosityLevel() const noexcept = 0;
 
 virtual void addVerbosity() noexcept = 0;
 
 virtual void reduceVerbosity() noexcept = 0;
 
 virtual void setVerbosityLevel(Verbosity) noexcept = 0;
 
 virtual char const* getTextFileName() const noexcept = 0;
 
 virtual void setTextFileName(char const* textFileName) noexcept = 0;
 
 virtual char const* getFullTextFileName() const noexcept = 0;
 
 virtual void setFullTextFileName(char const* fullTextFileName) noexcept = 0;
 
 virtual bool getPrintLayerInfo() const noexcept = 0;
 
 virtual void setPrintLayerInfo(bool) noexcept = 0;
 
}; // class IOnnxConfig
 
TENSORRTAPI IOnnxConfig* createONNXConfig();
 
} // namespace nvonnxparser
 
#endif