16 #include "core/graph/model.h"
18 namespace onnxruntime {
20 #pragma GCC diagnostic push
25 ORTInvoker(std::shared_ptr<IExecutionProvider> execution_provider,
27 const IOnnxRuntimeOpSchemaRegistryList& custom_op_registries) :
28 execution_provider_(std::move(execution_provider)), logger_(logger), custom_op_registries_(custom_op_registries) {
29 if (!execution_provider_) {
30 ORT_THROW(
"Execution provider is nullptr");
35 return *execution_provider_;
40 const std::vector<OrtValue>& inputs,
41 std::vector<OrtValue>& outputs,
47 std::shared_ptr<IExecutionProvider> execution_provider_;
51 const IOnnxRuntimeOpSchemaRegistryList& custom_op_registries_;
55 #pragma GCC diagnostic pop
GLsizei const GLchar *const * string
ORTInvoker(std::shared_ptr< IExecutionProvider > execution_provider, const logging::Logger &logger, const IOnnxRuntimeOpSchemaRegistryList &custom_op_registries)
std::unordered_map< std::string, ONNX_NAMESPACE::AttributeProto > NodeAttributes
common::Status Invoke(const std::string &op_name, const std::vector< OrtValue > &inputs, std::vector< OrtValue > &outputs, const NodeAttributes *attributes, const std::string &domain=kOnnxDomain, const int version=-1)
constexpr const char * kOnnxDomain
GT_API const UT_StringHolder version
IExecutionProvider & GetCurrentExecutionProvider()