OnnxRuntime
|
This is the complete list of members for Ort::CustomOpApi, including all inherited members.
CopyKernelInfo(const OrtKernelInfo *info) | Ort::CustomOpApi | |
CreateOp(const OrtKernelInfo *info, const char *op_name, const char *domain, int version, const char **type_constraint_names, const ONNXTensorElementDataType *type_constraint_values, int type_constraint_count, const OrtOpAttr *const *attr_values, int attr_count, int input_count, int output_count) | Ort::CustomOpApi | |
CreateOpAttr(const char *name, const void *data, int len, OrtOpAttrType type) | Ort::CustomOpApi | |
CustomOpApi(const OrtApi &api) | Ort::CustomOpApi | inline |
GetDimensions(const OrtTensorTypeAndShapeInfo *info, int64_t *dim_values, size_t dim_values_length) | Ort::CustomOpApi | |
GetDimensionsCount(const OrtTensorTypeAndShapeInfo *info) | Ort::CustomOpApi | |
GetTensorData(const OrtValue *value) | Ort::CustomOpApi | |
GetTensorElementType(const OrtTensorTypeAndShapeInfo *info) | Ort::CustomOpApi | |
GetTensorMemoryInfo(const OrtValue *value) | Ort::CustomOpApi | |
GetTensorMutableData(OrtValue *value) | Ort::CustomOpApi | |
GetTensorShape(const OrtTensorTypeAndShapeInfo *info) | Ort::CustomOpApi | |
GetTensorShapeElementCount(const OrtTensorTypeAndShapeInfo *info) | Ort::CustomOpApi | |
GetTensorTypeAndShape(const OrtValue *value) | Ort::CustomOpApi | |
InvokeOp(const OrtKernelContext *context, const OrtOp *ort_op, const OrtValue *const *input_values, int input_count, OrtValue *const *output_values, int output_count) | Ort::CustomOpApi | |
KernelContext_GetGPUComputeStream(const OrtKernelContext *context) | Ort::CustomOpApi | |
KernelContext_GetInput(const OrtKernelContext *context, size_t index) | Ort::CustomOpApi | |
KernelContext_GetInputCount(const OrtKernelContext *context) | Ort::CustomOpApi | |
KernelContext_GetOutput(OrtKernelContext *context, size_t index, const int64_t *dim_values, size_t dim_count) | Ort::CustomOpApi | |
KernelContext_GetOutputCount(const OrtKernelContext *context) | Ort::CustomOpApi | |
KernelInfoGetAttribute(const OrtKernelInfo *info, const char *name) | Ort::CustomOpApi | |
ReleaseKernelInfo(OrtKernelInfo *info_copy) | Ort::CustomOpApi | |
ReleaseOp(OrtOp *ort_op) | Ort::CustomOpApi | |
ReleaseOpAttr(OrtOpAttr *op_attr) | Ort::CustomOpApi | |
ReleaseTensorTypeAndShapeInfo(OrtTensorTypeAndShapeInfo *input) | Ort::CustomOpApi | |
SetDimensions(OrtTensorTypeAndShapeInfo *info, const int64_t *dim_values, size_t dim_count) | Ort::CustomOpApi | |
ThrowOnError(OrtStatus *result) | Ort::CustomOpApi |