ONNX Runtime
Loading...
Searching...
No Matches
OrtCustomOp Struct Reference

#include <onnxruntime_c_api.h>

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

Public Attributes

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)
 
OrtStatusPtr(* InferOutputShapeFn )(const struct OrtCustomOp *op, OrtShapeInferContext *)
 
int(* GetStartVersion )(const struct OrtCustomOp *op)
 
int(* GetEndVersion )(const struct OrtCustomOp *op)
 
size_t(* GetMayInplace )(int **input_index, int **output_index)
 

Member Data Documentation

◆ CreateKernel

void *( * OrtCustomOp::CreateKernel) (const struct OrtCustomOp *op, const OrtApi *api, const OrtKernelInfo *info)

◆ CreateKernelV2

OrtStatusPtr( * OrtCustomOp::CreateKernelV2) (const struct OrtCustomOp *op, const OrtApi *api, const OrtKernelInfo *info, void **kernel)

◆ GetEndVersion

int( * OrtCustomOp::GetEndVersion) (const struct OrtCustomOp *op)

◆ GetExecutionProviderType

const char *( * OrtCustomOp::GetExecutionProviderType) (const struct OrtCustomOp *op)

◆ GetInputCharacteristic

OrtCustomOpInputOutputCharacteristic( * OrtCustomOp::GetInputCharacteristic) (const struct OrtCustomOp *op, size_t index)

◆ GetInputMemoryType

OrtMemType( * OrtCustomOp::GetInputMemoryType) (const struct OrtCustomOp *op, size_t index)

◆ GetInputType

ONNXTensorElementDataType( * OrtCustomOp::GetInputType) (const struct OrtCustomOp *op, size_t index)

◆ GetInputTypeCount

size_t( * OrtCustomOp::GetInputTypeCount) (const struct OrtCustomOp *op)

◆ GetMayInplace

size_t( * OrtCustomOp::GetMayInplace) (int **input_index, int **output_index)

◆ GetName

const char *( * OrtCustomOp::GetName) (const struct OrtCustomOp *op)

◆ GetOutputCharacteristic

OrtCustomOpInputOutputCharacteristic( * OrtCustomOp::GetOutputCharacteristic) (const struct OrtCustomOp *op, size_t index)

◆ GetOutputType

ONNXTensorElementDataType( * OrtCustomOp::GetOutputType) (const struct OrtCustomOp *op, size_t index)

◆ GetOutputTypeCount

size_t( * OrtCustomOp::GetOutputTypeCount) (const struct OrtCustomOp *op)

◆ GetStartVersion

int( * OrtCustomOp::GetStartVersion) (const struct OrtCustomOp *op)

◆ GetVariadicInputHomogeneity

int( * OrtCustomOp::GetVariadicInputHomogeneity) (const struct OrtCustomOp *op)

◆ GetVariadicInputMinArity

int( * OrtCustomOp::GetVariadicInputMinArity) (const struct OrtCustomOp *op)

◆ GetVariadicOutputHomogeneity

int( * OrtCustomOp::GetVariadicOutputHomogeneity) (const struct OrtCustomOp *op)

◆ GetVariadicOutputMinArity

int( * OrtCustomOp::GetVariadicOutputMinArity) (const struct OrtCustomOp *op)

◆ InferOutputShapeFn

OrtStatusPtr( * OrtCustomOp::InferOutputShapeFn) (const struct OrtCustomOp *op, OrtShapeInferContext *)

◆ KernelCompute

void( * OrtCustomOp::KernelCompute) (void *op_kernel, OrtKernelContext *context)

◆ KernelComputeV2

OrtStatusPtr( * OrtCustomOp::KernelComputeV2) (void *op_kernel, OrtKernelContext *context)

◆ KernelDestroy

void( * OrtCustomOp::KernelDestroy) (void *op_kernel)

◆ version

uint32_t OrtCustomOp::version