![]() |
ONNX Runtime
|
Builder for OrtKernelDef. More...
#include <onnxruntime_cxx_api.h>
Public Member Functions | |
| KernelDefBuilder () | |
| Wraps OrtEpApi::CreateKernelDefBuilder. | |
| KernelDefBuilder (std::nullptr_t) | |
| Create an empty object, must be assigned a valid one to be used. | |
| KernelDefBuilder (OrtKernelDefBuilder *ort_kernel_def_builder) | |
| KernelDefBuilder & | SetOperatorType (const char *op_type) |
| KernelDefBuilder & | SetDomain (const char *domain) |
| KernelDefBuilder & | SetSinceVersion (int since_version_start, int since_version_end) |
| KernelDefBuilder & | SetExecutionProvider (const char *ep_name) |
| KernelDefBuilder & | SetInputMemType (size_t input_index, OrtMemType mem_type) |
| KernelDefBuilder & | SetOutputMemType (size_t output_index, OrtMemType mem_type) |
| KernelDefBuilder & | AddTypeConstraint (const char *arg_name, const OrtDataType *data_type) |
| KernelDefBuilder & | AddTypeConstraint (const char *arg_name, const std::vector< const OrtDataType * > &data_types) |
| KernelDefBuilder & | AddInputOutputAlias (int input_index, int output_index) |
| KernelDefBuilder & | AddInputOutputAliases (const std::vector< int > &input_indices, const std::vector< int > &output_indices) |
| KernelDefBuilder & | AddInputOutputMutableAlias (int input_index, int output_index) |
| KernelDefBuilder & | AddInputOutputMutableAliases (const std::vector< int > &input_indices, const std::vector< int > &output_indices) |
| KernelDef | Build () |
Public Member Functions inherited from Ort::detail::Base< OrtKernelDefBuilder > | |
| constexpr | Base ()=default |
| constexpr | Base (contained_type *p) noexcept |
| Base (const Base &)=delete | |
| Base (Base &&v) noexcept | |
| ~Base () | |
| Base & | operator= (const Base &)=delete |
| Base & | operator= (Base &&v) noexcept |
| constexpr | operator contained_type * () const noexcept |
| constexpr contained_type & | operator* () const noexcept |
| contained_type * | release () |
| Relinquishes ownership of the contained C object pointer The underlying object is not destroyed. | |
Additional Inherited Members | |
Public Types inherited from Ort::detail::Base< OrtKernelDefBuilder > | |
| using | contained_type = OrtKernelDefBuilder |
Protected Attributes inherited from Ort::detail::Base< OrtKernelDefBuilder > | |
| contained_type * | p_ |
Builder for OrtKernelDef.
Used by plugin EPs to build a kernel definition.
| Ort::KernelDefBuilder::KernelDefBuilder | ( | ) |
|
inlineexplicit |
Create an empty object, must be assigned a valid one to be used.
|
explicit |
| KernelDefBuilder & Ort::KernelDefBuilder::AddInputOutputAlias | ( | int | input_index, |
| int | output_index | ||
| ) |
| KernelDefBuilder & Ort::KernelDefBuilder::AddInputOutputAliases | ( | const std::vector< int > & | input_indices, |
| const std::vector< int > & | output_indices | ||
| ) |
| KernelDefBuilder & Ort::KernelDefBuilder::AddInputOutputMutableAlias | ( | int | input_index, |
| int | output_index | ||
| ) |
| KernelDefBuilder & Ort::KernelDefBuilder::AddInputOutputMutableAliases | ( | const std::vector< int > & | input_indices, |
| const std::vector< int > & | output_indices | ||
| ) |
| KernelDefBuilder & Ort::KernelDefBuilder::AddTypeConstraint | ( | const char * | arg_name, |
| const OrtDataType * | data_type | ||
| ) |
| KernelDefBuilder & Ort::KernelDefBuilder::AddTypeConstraint | ( | const char * | arg_name, |
| const std::vector< const OrtDataType * > & | data_types | ||
| ) |
| KernelDef Ort::KernelDefBuilder::Build | ( | ) |
| KernelDefBuilder & Ort::KernelDefBuilder::SetDomain | ( | const char * | domain | ) |
| KernelDefBuilder & Ort::KernelDefBuilder::SetExecutionProvider | ( | const char * | ep_name | ) |
| KernelDefBuilder & Ort::KernelDefBuilder::SetInputMemType | ( | size_t | input_index, |
| OrtMemType | mem_type | ||
| ) |
| KernelDefBuilder & Ort::KernelDefBuilder::SetOperatorType | ( | const char * | op_type | ) |
| KernelDefBuilder & Ort::KernelDefBuilder::SetOutputMemType | ( | size_t | output_index, |
| OrtMemType | mem_type | ||
| ) |
| KernelDefBuilder & Ort::KernelDefBuilder::SetSinceVersion | ( | int | since_version_start, |
| int | since_version_end | ||
| ) |