mirror of
https://github.com/saymrwulf/onnxruntime.git
synced 2026-06-03 23:49:44 +00:00
d761571afc
Update Python API to allow more flexibility for setting providers and provider options. The providers argument (InferenceSession/TrainingSession constructors, InferenceSession.set_providers()) now also accepts a tuple of (name, options dict). Fix get_available_providers() API (and the corresponding function in the C API) to return the providers in default priority order. Now it can be used as a starting point for the providers argument and maintain the default priority order. Convert some usages of the deprecated global configuration functions to use EP-specific options instead. Update some EP-specific option parsing to fail on unknown options. Other clean up.
38 lines
1.9 KiB
C++
38 lines
1.9 KiB
C++
// Copyright (c) Microsoft Corporation. All rights reserved.
|
|
// Licensed under the MIT License.
|
|
|
|
#pragma once
|
|
|
|
namespace onnxruntime {
|
|
|
|
constexpr const char* kNoOp = "NoOp";
|
|
constexpr const char* kConstant = "Constant";
|
|
constexpr const char* kFunctionOp = "_kFunctionOp";
|
|
constexpr const char* kConstantValue = "value";
|
|
constexpr const char* kOnnxDomain = "";
|
|
constexpr const char* kOnnxDomainAlias = "ai.onnx";
|
|
constexpr const char* kMLDomain = "ai.onnx.ml";
|
|
constexpr const char* kMSDomain = "com.microsoft";
|
|
constexpr const char* kMSNchwcDomain = "com.microsoft.nchwc";
|
|
constexpr const char* kMSFeaturizersDomain = "com.microsoft.mlfeaturizers";
|
|
constexpr const char* kMSDmlDomain = "com.microsoft.dml";
|
|
constexpr const char* kNGraphDomain = "com.intel.ai";
|
|
constexpr const char* kMIGraphXDomain = "";
|
|
constexpr const char* kVitisAIDomain = "com.xilinx";
|
|
|
|
constexpr const char* kCpuExecutionProvider = "CPUExecutionProvider";
|
|
constexpr const char* kCudaExecutionProvider = "CUDAExecutionProvider";
|
|
constexpr const char* kDnnlExecutionProvider = "DnnlExecutionProvider";
|
|
constexpr const char* kOpenVINOExecutionProvider = "OpenVINOExecutionProvider";
|
|
constexpr const char* kNupharExecutionProvider = "NupharExecutionProvider";
|
|
constexpr const char* kVitisAIExecutionProvider = "VitisAIExecutionProvider";
|
|
constexpr const char* kTensorrtExecutionProvider = "TensorrtExecutionProvider";
|
|
constexpr const char* kNnapiExecutionProvider = "NnapiExecutionProvider";
|
|
constexpr const char* kRknpuExecutionProvider = "RknpuExecutionProvider";
|
|
constexpr const char* kDmlExecutionProvider = "DmlExecutionProvider";
|
|
constexpr const char* kMIGraphXExecutionProvider = "MIGraphXExecutionProvider";
|
|
constexpr const char* kAclExecutionProvider = "ACLExecutionProvider";
|
|
constexpr const char* kArmNNExecutionProvider = "ArmNNExecutionProvider";
|
|
constexpr const char* kRocmExecutionProvider = "ROCMExecutionProvider";
|
|
|
|
} // namespace onnxruntime
|