HDK
 All Classes Namespaces Files Functions Variables Typedefs Enumerations Enumerator Friends Macros Groups Pages
onnxruntime::KernelDef Class Reference

#include <kernel_def_builder.h>

Public Member Functions

 KernelDef ()=default
 
const std::stringOpName () const
 
const std::stringDomain () const
 
void SinceVersion (int *start, int *end) const
 
onnxruntime::ProviderType Provider () const
 
const std::unordered_map
< std::string, std::vector
< MLDataType > > & 
TypeConstraints () const
 
const std::vector< std::pair
< int, int > > & 
MayInplace () const
 
const std::vector< std::pair
< int, int > > & 
Alias () const
 
const std::optional< std::pair
< int, int > > & 
VariadicAlias () const
 
OrtMemType InputMemoryType (size_t input_index) const
 
bool IsInputOnCpu (size_t input_index) const
 
bool IsOutputOnCpu (size_t output_index) const
 
bool AllocateInputsContiguously () const
 
bool HasExternalOutputs () const
 
OrtMemType OutputMemoryType (size_t output_index) const
 
int ExecQueueId () const
 
bool IsConflict (const KernelDef &other) const
 

Friends

class KernelDefBuilder
 

Detailed Description

Definition at line 23 of file kernel_def_builder.h.

Constructor & Destructor Documentation

onnxruntime::KernelDef::KernelDef ( )
explicitdefault

Member Function Documentation

const std::vector<std::pair<int, int> >& onnxruntime::KernelDef::Alias ( ) const
inline

Definition at line 65 of file kernel_def_builder.h.

bool onnxruntime::KernelDef::AllocateInputsContiguously ( ) const
inline

Definition at line 84 of file kernel_def_builder.h.

const std::string& onnxruntime::KernelDef::Domain ( ) const
inline

Definition at line 37 of file kernel_def_builder.h.

int onnxruntime::KernelDef::ExecQueueId ( ) const
inline

Definition at line 100 of file kernel_def_builder.h.

bool onnxruntime::KernelDef::HasExternalOutputs ( ) const
inline

Definition at line 86 of file kernel_def_builder.h.

OrtMemType onnxruntime::KernelDef::InputMemoryType ( size_t  input_index) const
inline

Definition at line 73 of file kernel_def_builder.h.

bool onnxruntime::KernelDef::IsConflict ( const KernelDef other) const
bool onnxruntime::KernelDef::IsInputOnCpu ( size_t  input_index) const
inline

Definition at line 80 of file kernel_def_builder.h.

bool onnxruntime::KernelDef::IsOutputOnCpu ( size_t  output_index) const
inline

Definition at line 82 of file kernel_def_builder.h.

const std::vector<std::pair<int, int> >& onnxruntime::KernelDef::MayInplace ( ) const
inline

Definition at line 61 of file kernel_def_builder.h.

const std::string& onnxruntime::KernelDef::OpName ( ) const
inline

Definition at line 33 of file kernel_def_builder.h.

OrtMemType onnxruntime::KernelDef::OutputMemoryType ( size_t  output_index) const
inline

Definition at line 93 of file kernel_def_builder.h.

onnxruntime::ProviderType onnxruntime::KernelDef::Provider ( ) const
inline

Definition at line 52 of file kernel_def_builder.h.

void onnxruntime::KernelDef::SinceVersion ( int start,
int end 
) const
inline

Definition at line 41 of file kernel_def_builder.h.

const std::unordered_map<std::string, std::vector<MLDataType> >& onnxruntime::KernelDef::TypeConstraints ( ) const
inline

Definition at line 57 of file kernel_def_builder.h.

const std::optional<std::pair<int, int> >& onnxruntime::KernelDef::VariadicAlias ( ) const
inline

Definition at line 69 of file kernel_def_builder.h.

Friends And Related Function Documentation

friend class KernelDefBuilder
friend

Definition at line 107 of file kernel_def_builder.h.


The documentation for this class was generated from the following file: