mirror of
https://github.com/saymrwulf/onnxruntime.git
synced 2026-06-08 00:23:03 +00:00
8d87fdcfa1
### Description Added APIs. ### Motivation and Context Addresses https://github.com/microsoft/onnxruntime/issues/14584 Cc: @Craigacp cp
27 lines
1.1 KiB
C++
27 lines
1.1 KiB
C++
// Copyright (c) Microsoft Corporation. All rights reserved.
|
|
// Licensed under the MIT License.
|
|
|
|
#include "onnxruntime_pybind.h" // must use this for the include of <pybind11/pybind11.h>
|
|
#include <pybind11/stl.h>
|
|
#include "core/providers/get_execution_providers.h"
|
|
#include "onnxruntime_config.h"
|
|
|
|
namespace onnxruntime {
|
|
namespace python {
|
|
namespace py = pybind11;
|
|
|
|
void CreateInferencePybindStateModule(py::module& m);
|
|
|
|
PYBIND11_MODULE(onnxruntime_pybind11_state, m) {
|
|
CreateInferencePybindStateModule(m);
|
|
// move it out of shared method since training build has a little different behavior.
|
|
m.def(
|
|
"get_available_providers", []() -> const std::vector<std::string>& { return GetAvailableExecutionProviderNames(); },
|
|
"Return list of available Execution Providers in this installed version of Onnxruntime. "
|
|
"The order of elements represents the default priority order of Execution Providers "
|
|
"from highest to lowest.");
|
|
|
|
m.def("get_version_string", []() -> std::string { return ORT_VERSION; });
|
|
}
|
|
} // namespace python
|
|
} // namespace onnxruntime
|