Polygraphy
This page includes the Python API reference documentation for Polygraphy. Polygraphy is a toolkit designed to assist in running and debugging deep learning models in various frameworks.
For installation instructions, examples, and information about the CLI tools, see the GitHub repository instead.
For a conceptual overview of the Python API, see this page.
Warning
Any APIs not documented here should be considered internal only and do not adhere to the deprecation policy for public APIs. Thus, they may be modified or removed at any time without warning. Avoid using undocumented APIs!
API Reference: Main
- Backends
- Base Interface
- Common
- ONNX
- Loaders
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()
- Loaders
- ONNX-Runtime
- Plugin Reference
- TensorFlow 1.X
- TensorRT
- Algorithm Selector
- Calibrator
- Config
- Loaders
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()BytesFromEnginebytes_from_engine()SaveEnginesave_engine()OnnxLikeFromNetworkonnx_like_from_network()MarkDebugmark_debug()
- Optimization Profile
- Runners
- Utilities
- Comparing Results
API Reference: Miscellaneous
- Common
- Global Configuration
- Constants
- CUDA Wrapper
- Data Types
- Exceptions
- Function Helpers
- JSON Utilities
- Logger
LogModeSeverityTrieLoggerLogger.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
API Reference: Development