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