ONNX Runtime
Loading...
Searching...
No Matches
Ort::KernelDefBuilder Struct Reference

Builder for OrtKernelDef. More...

#include <onnxruntime_cxx_api.h>

Inheritance diagram for Ort::KernelDefBuilder:
Ort::detail::Base< OrtKernelDefBuilder >

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)
 
KernelDefBuilderSetOperatorType (const char *op_type)
 
KernelDefBuilderSetDomain (const char *domain)
 
KernelDefBuilderSetSinceVersion (int since_version_start, int since_version_end)
 
KernelDefBuilderSetExecutionProvider (const char *ep_name)
 
KernelDefBuilderSetInputMemType (size_t input_index, OrtMemType mem_type)
 
KernelDefBuilderSetOutputMemType (size_t output_index, OrtMemType mem_type)
 
KernelDefBuilderAddTypeConstraint (const char *arg_name, const OrtDataType *data_type)
 
KernelDefBuilderAddTypeConstraint (const char *arg_name, const std::vector< const OrtDataType * > &data_types)
 
KernelDefBuilderAddInputOutputAlias (int input_index, int output_index)
 
KernelDefBuilderAddInputOutputAliases (const std::vector< int > &input_indices, const std::vector< int > &output_indices)
 
KernelDefBuilderAddInputOutputMutableAlias (int input_index, int output_index)
 
KernelDefBuilderAddInputOutputMutableAliases (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 ()
 
Baseoperator= (const Base &)=delete
 
Baseoperator= (Base &&v) noexcept
 
constexpr operator contained_type * () const noexcept
 
constexpr contained_typeoperator* () const noexcept
 
contained_typerelease ()
 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_typep_
 

Detailed Description

Builder for OrtKernelDef.

Used by plugin EPs to build a kernel definition.

Constructor & Destructor Documentation

◆ KernelDefBuilder() [1/3]

Ort::KernelDefBuilder::KernelDefBuilder ( )

◆ KernelDefBuilder() [2/3]

Ort::KernelDefBuilder::KernelDefBuilder ( std::nullptr_t  )
inlineexplicit

Create an empty object, must be assigned a valid one to be used.

◆ KernelDefBuilder() [3/3]

Ort::KernelDefBuilder::KernelDefBuilder ( OrtKernelDefBuilder *  ort_kernel_def_builder)
explicit

Member Function Documentation

◆ AddInputOutputAlias()

KernelDefBuilder & Ort::KernelDefBuilder::AddInputOutputAlias ( int  input_index,
int  output_index 
)

◆ AddInputOutputAliases()

KernelDefBuilder & Ort::KernelDefBuilder::AddInputOutputAliases ( const std::vector< int > &  input_indices,
const std::vector< int > &  output_indices 
)

◆ AddInputOutputMutableAlias()

KernelDefBuilder & Ort::KernelDefBuilder::AddInputOutputMutableAlias ( int  input_index,
int  output_index 
)

◆ AddInputOutputMutableAliases()

KernelDefBuilder & Ort::KernelDefBuilder::AddInputOutputMutableAliases ( const std::vector< int > &  input_indices,
const std::vector< int > &  output_indices 
)

◆ AddTypeConstraint() [1/2]

KernelDefBuilder & Ort::KernelDefBuilder::AddTypeConstraint ( const char *  arg_name,
const OrtDataType *  data_type 
)

◆ AddTypeConstraint() [2/2]

KernelDefBuilder & Ort::KernelDefBuilder::AddTypeConstraint ( const char *  arg_name,
const std::vector< const OrtDataType * > &  data_types 
)

◆ Build()

KernelDef Ort::KernelDefBuilder::Build ( )

◆ SetDomain()

KernelDefBuilder & Ort::KernelDefBuilder::SetDomain ( const char *  domain)

◆ SetExecutionProvider()

KernelDefBuilder & Ort::KernelDefBuilder::SetExecutionProvider ( const char *  ep_name)

◆ SetInputMemType()

KernelDefBuilder & Ort::KernelDefBuilder::SetInputMemType ( size_t  input_index,
OrtMemType  mem_type 
)

◆ SetOperatorType()

KernelDefBuilder & Ort::KernelDefBuilder::SetOperatorType ( const char *  op_type)

◆ SetOutputMemType()

KernelDefBuilder & Ort::KernelDefBuilder::SetOutputMemType ( size_t  output_index,
OrtMemType  mem_type 
)

◆ SetSinceVersion()

KernelDefBuilder & Ort::KernelDefBuilder::SetSinceVersion ( int  since_version_start,
int  since_version_end 
)