TensorRT
模块: polygraphy.backend.trt
- 算法选择器
- 校准器
- 配置
- 加载器
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
- 运行器
- 实用工具