ONNX Runtime
Loading...
Searching...
No Matches
Ort::CustomOpBase< TOp, TKernel, WithStatus > Struct Template Reference

#include <onnxruntime_cxx_api.h>

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

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::string > GetSessionConfigKeys () const
 

Protected Member Functions

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

Additional Inherited Members

- Public Attributes inherited from OrtCustomOp
uint32_t version
 
void *(* CreateKernel )(const struct OrtCustomOp *op, const OrtApi *api, const OrtKernelInfo *info)
 
const char *(* GetName )(const struct OrtCustomOp *op)
 
const char *(* GetExecutionProviderType )(const struct OrtCustomOp *op)
 
ONNXTensorElementDataType(* GetInputType )(const struct OrtCustomOp *op, size_t index)
 
size_t(* GetInputTypeCount )(const struct OrtCustomOp *op)
 
ONNXTensorElementDataType(* GetOutputType )(const struct OrtCustomOp *op, size_t index)
 
size_t(* GetOutputTypeCount )(const struct OrtCustomOp *op)
 
void(* KernelCompute )(void *op_kernel, OrtKernelContext *context)
 
void(* KernelDestroy )(void *op_kernel)
 
OrtCustomOpInputOutputCharacteristic(* GetInputCharacteristic )(const struct OrtCustomOp *op, size_t index)
 
OrtCustomOpInputOutputCharacteristic(* GetOutputCharacteristic )(const struct OrtCustomOp *op, size_t index)
 
OrtMemType(* GetInputMemoryType )(const struct OrtCustomOp *op, size_t index)
 
int(* GetVariadicInputMinArity )(const struct OrtCustomOp *op)
 
int(* GetVariadicInputHomogeneity )(const struct OrtCustomOp *op)
 
int(* GetVariadicOutputMinArity )(const struct OrtCustomOp *op)
 
int(* GetVariadicOutputHomogeneity )(const struct OrtCustomOp *op)
 
OrtStatusPtr(* CreateKernelV2 )(const struct OrtCustomOp *op, const OrtApi *api, const OrtKernelInfo *info, void **kernel)
 
OrtStatusPtr(* KernelComputeV2 )(void *op_kernel, OrtKernelContext *context)
 

Constructor & Destructor Documentation

◆ CustomOpBase()

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

Member Function Documentation

◆ GetExecutionProviderType()

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

◆ GetInputCharacteristic()

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

◆ GetInputMemoryType()

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

◆ GetOutputCharacteristic()

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

◆ GetSessionConfigKeys()

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

◆ GetSessionConfigs()

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

◆ GetVariadicInputHomogeneity()

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

◆ GetVariadicInputMinArity()

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

◆ GetVariadicOutputHomogeneity()

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

◆ GetVariadicOutputMinArity()

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