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