OnnxRuntime
|
#include <onnxruntime_cxx_api.h>
Inherits OrtCustomOp.
Public Member Functions | |
CustomOpBase () | |
const char * | GetExecutionProviderType () const |
OrtCustomOpInputOutputCharacteristic | GetInputCharacteristic (size_t) const |
OrtCustomOpInputOutputCharacteristic | GetOutputCharacteristic (size_t) 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) |
|
inline |
|
inline |
|
inline |
|
inline |