OnnxRuntime
|
#include <onnxruntime_cxx_api.h>
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) |
|
inline |
|
inline |
|
inline |
|
inline |
|
inline |
|
inline |
|
protected |
|
inline |
|
inline |
|
inline |
|
inline |