NvInferSerialize.h

前往此文件的文档。

/*
 * SPDX-FileCopyrightText: 版权所有 (c) 1993-2024 NVIDIA CORPORATION & AFFILIATES。保留所有权利。
 * SPDX-许可证标识符: Apache-2.0
 *
 * 根据 Apache 许可证 2.0 版获得许可（"许可证"）；
 * 除非遵守许可证，否则您不得使用此文件。
 * 您可以在以下位置获取许可证副本：
 *
 * https://apache.ac.cn/licenses/LICENSE-2.0
 *
 * 除非适用法律要求或以书面形式达成一致，否则
 * 根据本许可证分发的软件以“现状”为基础分发，
 * 不附带任何形式的明示或暗示的保证或条件。
 * 请参阅许可证，了解有关权限和
 * 许可证下的限制的具体语言。
 */
 
#ifndef NV_INFER_SERIALIZE_H
#define NV_INFER_SERIALIZE_H
 
#include "NvInfer.h"
 
extern "C" TENSORRTAPI void* serializeNetwork_INTERNAL(void const* network) noexcept;
 
namespace nvinfer1
{
namespace serialize
{
 
inline nvinfer1::IHostMemory* serializeNetwork(nvinfer1::INetworkDefinition const& network) noexcept
{
 return static_cast<nvinfer1::IHostMemory*>(serializeNetwork_INTERNAL(&network));
}
 
} // namespace serialize
} // namespace nvinfer1
 
#endif // NV_INFER_SERIALIZE_H