Polygraphy
此页面包含 Polygraphy 的 Python API 参考文档。Polygraphy 是一个工具包,旨在帮助在各种框架中运行和调试深度学习模型。
有关安装说明、示例以及 CLI 工具的信息,请参阅GitHub 仓库。
有关 Python API 的概念概述,请参阅此页面。
警告
任何此处未记录的 API 都应被视为仅供内部使用,并且不遵守公共 API 的弃用策略。因此,它们可能会在任何时候被修改或删除,恕不另行通知。避免使用未记录的 API!
API 参考:主要
- 后端
- 基础接口
- 常用
- ONNX
- 加载器
GsFromOnnx
gs_from_onnx()
OnnxFromPath
onnx_from_path()
OnnxFromTfGraph
onnx_from_tf_graph()
ModifyOutputs
modify_outputs()
ConvertToFp16
convert_to_fp16()
FoldConstants
fold_constants()
SetUpperBound
set_upper_bound()
InferShapes
infer_shapes()
ExtractSubgraph
extract_subgraph()
SaveOnnx
save_onnx()
BytesFromOnnx
bytes_from_onnx()
OnnxFromBytes
onnx_from_bytes()
- 加载器
- ONNX-Runtime
- 插件参考
- TensorFlow 1.X
- TensorRT
- 算法选择器
- Calibrator
- Config
- 加载器
LoadPlugins
load_plugins()
CreateNetwork
create_network()
NetworkFromOnnxBytes
network_from_onnx_bytes()
NetworkFromOnnxPath
network_from_onnx_path()
PostprocessNetwork
postprocess_network()
ModifyNetworkOutputs
modify_network_outputs()
SetLayerPrecisions
set_layer_precisions()
SetTensorDatatypes
set_tensor_datatypes()
SetTensorFormats
set_tensor_formats()
LoadRuntime
load_runtime()
EngineBytesFromNetwork
engine_bytes_from_network()
EngineFromNetwork
engine_from_network()
EngineFromBytes
engine_from_bytes()
EngineFromPath
engine_from_path()
BytesFromEngine
bytes_from_engine()
BufferFromEngine
buffer_from_engine()
SaveEngine
save_engine()
OnnxLikeFromNetwork
onnx_like_from_network()
MarkDebug
mark_debug()
- 优化 Profile
- 运行器
- 实用工具
- 比较结果
API 参考:其他
- 常用
- 全局配置
- 常量
- CUDA 封装器
- 数据类型
- 异常
- 函数助手
- JSON 工具
- 日志记录器
LogMode
SeverityTrie
日志记录器
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_file
Logger.module_severity
Logger.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