后端
模块: polygraphy.backend
- 基础接口
- 通用
- 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
- 运行器
- 实用工具