HDK
 All Classes Namespaces Files Functions Variables Typedefs Enumerations Enumerator Friends Macros Groups Pages
Ort::CustomOpBase< TOp, TKernel, WithStatus > Struct Template Reference

#include <onnxruntime_cxx_api.h>

+ Inheritance diagram for Ort::CustomOpBase< TOp, TKernel, WithStatus >:

Public Member Functions

 CustomOpBase ()
 
const char * GetExecutionProviderType () const
 
OrtCustomOpInputOutputCharacteristic GetInputCharacteristic (size_t) const
 
OrtCustomOpInputOutputCharacteristic GetOutputCharacteristic (size_t) const
 
OrtMemType GetInputMemoryType (size_t) const
 
int GetVariadicInputMinArity () const
 
bool GetVariadicInputHomogeneity () const
 
int GetVariadicOutputMinArity () const
 
bool GetVariadicOutputHomogeneity () const
 
std::vector< std::stringGetSessionConfigKeys () const
 
template<typename C >
void SetShapeInferFn (...)
 

Public Attributes

template<typename C >
decltype(&C::InferOutputShape) SetShapeInferFn (decltype(&C::InferOutputShape))
 

Protected Member Functions

void GetSessionConfigs (std::unordered_map< std::string, std::string > &out, ConstSessionOptions options) const
 

Protected Attributes

int start_ver_ = 1
 
int end_ver_ = MAX_CUSTOM_OP_END_VER
 

Detailed Description

template<typename TOp, typename TKernel, bool WithStatus = false>
struct Ort::CustomOpBase< TOp, TKernel, WithStatus >

Definition at line 2238 of file onnxruntime_cxx_api.h.

Constructor & Destructor Documentation

template<typename TOp , typename TKernel , bool WithStatus = false>
Ort::CustomOpBase< TOp, TKernel, WithStatus >::CustomOpBase ( )
inline

Definition at line 2239 of file onnxruntime_cxx_api.h.

Member Function Documentation

template<typename TOp , typename TKernel , bool WithStatus = false>
const char* Ort::CustomOpBase< TOp, TKernel, WithStatus >::GetExecutionProviderType ( ) const
inline

Definition at line 2300 of file onnxruntime_cxx_api.h.

template<typename TOp , typename TKernel , bool WithStatus = false>
OrtCustomOpInputOutputCharacteristic Ort::CustomOpBase< TOp, TKernel, WithStatus >::GetInputCharacteristic ( size_t  ) const
inline

Definition at line 2304 of file onnxruntime_cxx_api.h.

template<typename TOp , typename TKernel , bool WithStatus = false>
OrtMemType Ort::CustomOpBase< TOp, TKernel, WithStatus >::GetInputMemoryType ( size_t  ) const
inline

Definition at line 2313 of file onnxruntime_cxx_api.h.

template<typename TOp , typename TKernel , bool WithStatus = false>
OrtCustomOpInputOutputCharacteristic Ort::CustomOpBase< TOp, TKernel, WithStatus >::GetOutputCharacteristic ( size_t  ) const
inline

Definition at line 2308 of file onnxruntime_cxx_api.h.

template<typename TOp , typename TKernel , bool WithStatus = false>
std::vector<std::string> Ort::CustomOpBase< TOp, TKernel, WithStatus >::GetSessionConfigKeys ( ) const
inline

Definition at line 2344 of file onnxruntime_cxx_api.h.

template<typename TOp , typename TKernel , bool WithStatus = false>
void Ort::CustomOpBase< TOp, TKernel, WithStatus >::GetSessionConfigs ( std::unordered_map< std::string, std::string > &  out,
ConstSessionOptions  options 
) const
protected
template<typename TOp , typename TKernel , bool WithStatus = false>
bool Ort::CustomOpBase< TOp, TKernel, WithStatus >::GetVariadicInputHomogeneity ( ) const
inline

Definition at line 2325 of file onnxruntime_cxx_api.h.

template<typename TOp , typename TKernel , bool WithStatus = false>
int Ort::CustomOpBase< TOp, TKernel, WithStatus >::GetVariadicInputMinArity ( ) const
inline

Definition at line 2319 of file onnxruntime_cxx_api.h.

template<typename TOp , typename TKernel , bool WithStatus = false>
bool Ort::CustomOpBase< TOp, TKernel, WithStatus >::GetVariadicOutputHomogeneity ( ) const
inline

Definition at line 2337 of file onnxruntime_cxx_api.h.

template<typename TOp , typename TKernel , bool WithStatus = false>
int Ort::CustomOpBase< TOp, TKernel, WithStatus >::GetVariadicOutputMinArity ( ) const
inline

Definition at line 2331 of file onnxruntime_cxx_api.h.

template<typename TOp , typename TKernel , bool WithStatus = false>
template<typename C >
void Ort::CustomOpBase< TOp, TKernel, WithStatus >::SetShapeInferFn (   ...)
inline

Definition at line 2358 of file onnxruntime_cxx_api.h.

Member Data Documentation

template<typename TOp , typename TKernel , bool WithStatus = false>
int Ort::CustomOpBase< TOp, TKernel, WithStatus >::end_ver_ = MAX_CUSTOM_OP_END_VER
protected

Definition at line 2367 of file onnxruntime_cxx_api.h.

template<typename TOp , typename TKernel , bool WithStatus = false>
template<typename C >
decltype(&C::InferOutputShape) Ort::CustomOpBase< TOp, TKernel, WithStatus >::SetShapeInferFn(decltype(&C::InferOutputShape))
inline

Definition at line 2349 of file onnxruntime_cxx_api.h.

template<typename TOp , typename TKernel , bool WithStatus = false>
int Ort::CustomOpBase< TOp, TKernel, WithStatus >::start_ver_ = 1
protected

Definition at line 2366 of file onnxruntime_cxx_api.h.


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