HDK
 All Classes Namespaces Files Functions Variables Typedefs Enumerations Enumerator Friends Macros Groups Pages
onnxruntime::ORTInvoker Class Reference

#include <ort_kernel_invoker.h>

Public Member Functions

 ORTInvoker (std::shared_ptr< IExecutionProvider > execution_provider, const logging::Logger &logger, const IOnnxRuntimeOpSchemaRegistryList &custom_op_registries)
 
IExecutionProviderGetCurrentExecutionProvider ()
 
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)
 

Detailed Description

Definition at line 23 of file ort_kernel_invoker.h.

Constructor & Destructor Documentation

onnxruntime::ORTInvoker::ORTInvoker ( std::shared_ptr< IExecutionProvider execution_provider,
const logging::Logger logger,
const IOnnxRuntimeOpSchemaRegistryList &  custom_op_registries 
)
inline

Definition at line 25 of file ort_kernel_invoker.h.

Member Function Documentation

IExecutionProvider& onnxruntime::ORTInvoker::GetCurrentExecutionProvider ( )
inline

Definition at line 33 of file ort_kernel_invoker.h.

common::Status onnxruntime::ORTInvoker::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 
)

The documentation for this class was generated from the following file: