mirror of
https://github.com/saymrwulf/onnxruntime.git
synced 2026-05-16 21:00:14 +00:00
Two major modifications of this PR: 1. Refactor OrtTensorRTProviderOptions initialization and make it easy to add new field. 2. Make Python API capable of using TensorRT plugins by adding new Python binding api `register_tensorrt_plugins_as_custom_ops`. (It needs to register ep's custom op domain before model load. For C++ API, it's slightly different, when calling SessionOptionsAppendExecutionProvider_TensorRT_XX, it appends cutom op domain to session option. Later ORT can register custom op domain from session option before model loading) |
||
|---|---|---|
| .. | ||
| common.h | ||
| directml_load_helper.cc | ||
| directml_load_helper.h | ||
| inference_session_wrap.cc | ||
| inference_session_wrap.h | ||
| main.cc | ||
| run_options_helper.cc | ||
| run_options_helper.h | ||
| session_options_helper.cc | ||
| session_options_helper.h | ||
| tensor_helper.cc | ||
| tensor_helper.h | ||