Polygraphy
此页面包含 Polygraphy 的 Python API 参考文档。Polygraphy 是一个工具包,旨在帮助在各种框架中运行和调试深度学习模型。
有关安装说明、示例以及 CLI 工具的信息,请参阅GitHub 仓库。
有关 Python API 的概念概述,请参阅此页面。
警告
任何此处未记录的 API 都应被视为仅供内部使用,并且不遵守公共 API 的弃用策略。因此,它们可能会在任何时候被修改或删除,恕不另行通知。避免使用未记录的 API!
API 参考:主要
- 后端
- 基础接口
- 常用
- ONNX
- 加载器
GsFromOnnxgs_from_onnx()OnnxFromPathonnx_from_path()OnnxFromTfGraphonnx_from_tf_graph()ModifyOutputsmodify_outputs()ConvertToFp16convert_to_fp16()FoldConstantsfold_constants()SetUpperBoundset_upper_bound()InferShapesinfer_shapes()ExtractSubgraphextract_subgraph()SaveOnnxsave_onnx()BytesFromOnnxbytes_from_onnx()OnnxFromBytesonnx_from_bytes()
- 加载器
- ONNX-Runtime
- 插件参考
- TensorFlow 1.X
- TensorRT
- 算法选择器
- Calibrator
- Config
- 加载器
LoadPluginsload_plugins()CreateNetworkcreate_network()NetworkFromOnnxBytesnetwork_from_onnx_bytes()NetworkFromOnnxPathnetwork_from_onnx_path()PostprocessNetworkpostprocess_network()ModifyNetworkOutputsmodify_network_outputs()SetLayerPrecisionsset_layer_precisions()SetTensorDatatypesset_tensor_datatypes()SetTensorFormatsset_tensor_formats()LoadRuntimeload_runtime()EngineBytesFromNetworkengine_bytes_from_network()EngineFromNetworkengine_from_network()EngineFromBytesengine_from_bytes()EngineFromPathengine_from_path()BytesFromEnginebytes_from_engine()BufferFromEnginebuffer_from_engine()SaveEnginesave_engine()OnnxLikeFromNetworkonnx_like_from_network()MarkDebugmark_debug()
- 优化 Profile
- 运行器
- 实用工具
- 比较结果
API 参考:其他
- 常用
- 全局配置
- 常量
- CUDA 封装器
- 数据类型
- 异常
- 函数助手
- JSON 工具
- 日志记录器
LogModeSeverityTrie日志记录器Logger.ULTRA_VERBOSELogger.SUPER_VERBOSELogger.EXTRA_VERBOSELogger.VERBOSELogger.INFOLogger.STARTLogger.FINISHLogger.WARNINGLogger.ERRORLogger.CRITICALLogger.log_fileLogger.module_severityLogger.module_path()Logger.register_callback()Logger.indent()Logger.verbosity()Logger.log()Logger.ultra_verbose()Logger.super_verbose()Logger.extra_verbose()Logger.verbose()Logger.info()Logger.start()Logger.finish()Logger.warning()Logger.error()Logger.critical()Logger.log_exception()
G_LOGGER