mirror of
https://github.com/saymrwulf/onnxruntime.git
synced 2026-06-28 03:20:58 +00:00
569876fb16
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)
39 lines
1 KiB
C
39 lines
1 KiB
C
// Copyright (c) Microsoft Corporation. All rights reserved.
|
|
// Licensed under the MIT License.
|
|
|
|
#pragma once
|
|
#include "core/providers/cpu/cpu_provider_factory.h"
|
|
|
|
#ifdef USE_DNNL
|
|
#include "core/providers/dnnl/dnnl_provider_factory.h"
|
|
#endif
|
|
#ifdef USE_TVM
|
|
#include "core/providers/tvm/tvm_provider_factory.h"
|
|
#endif
|
|
#ifdef USE_OPENVINO
|
|
#include "core/providers/openvino/openvino_provider_factory.h"
|
|
#endif
|
|
#ifdef USE_NNAPI
|
|
#include "core/providers/nnapi/nnapi_provider_factory.h"
|
|
#endif
|
|
#ifdef USE_COREML
|
|
#include "core/providers/coreml/coreml_provider_factory.h"
|
|
#endif
|
|
#ifdef USE_DML
|
|
#include "core/providers/dml/dml_provider_factory.h"
|
|
#endif
|
|
#ifdef USE_ACL
|
|
#include "core/providers/acl/acl_provider_factory.h"
|
|
#endif
|
|
#ifdef USE_ARMNN
|
|
#include "core/providers/armnn/armnn_provider_factory.h"
|
|
#endif
|
|
#ifdef USE_MIGRAPHX
|
|
#include "core/providers/migraphx/migraphx_provider_factory.h"
|
|
#endif
|
|
#ifdef USE_XNNPACK
|
|
#include "core/providers/xnnpack/xnnpack_provider_factory_creator.h"
|
|
#endif
|
|
#ifdef USE_CANN
|
|
#include "core/providers/cann/cann_provider_factory.h"
|
|
#endif
|