Public Member Functions
	KernelRegistry ()=default

Status	Register (KernelDefBuilder &kernel_def_builder, const KernelCreateFn &kernel_creator)

Status	Register (KernelCreateInfo &&create_info)

Status	TryFindKernel (const Node &node, ProviderType exec_provider, const IKernelTypeStrResolver &kernel_type_str_resolver, const KernelCreateInfo **out) const

Status	TryFindKernel (const std::string &op_name, const std::string &domain, const int &version, const std::unordered_map< std::string, MLDataType > &type_constraints, ProviderType exec_provider, const KernelCreateInfo **out) const

bool	IsEmpty () const

Static Public Member Functions
static bool	HasImplementationOf (const KernelRegistry &r, const Node &node, ProviderType exec_provider, const IKernelTypeStrResolver &kernel_type_str_resolver)

Detailed Description

Each provider has a KernelRegistry. Often, the KernelRegistry only belongs to that specific provider.

Definition at line 20 of file kernel_registry.h.

Constructor & Destructor Documentation

onnxruntime::KernelRegistry::KernelRegistry ( )

default

static bool onnxruntime::KernelRegistry::HasImplementationOf	(	const KernelRegistry &	r,
		const Node &	node,
		ProviderType	exec_provider,
		const IKernelTypeStrResolver &	kernel_type_str_resolver
	)

inlinestatic

Definition at line 37 of file kernel_registry.h.

bool onnxruntime::KernelRegistry::IsEmpty ( ) const

inline

Definition at line 52 of file kernel_registry.h.

Status onnxruntime::KernelRegistry::Register	(	KernelDefBuilder &	kernel_def_builder,
		const KernelCreateFn &	kernel_creator
	)

Status onnxruntime::KernelRegistry::Register ( KernelCreateInfo && create_info )

Status onnxruntime::KernelRegistry::TryFindKernel	(	const Node &	node,
		ProviderType	exec_provider,
		const IKernelTypeStrResolver &	kernel_type_str_resolver,
		const KernelCreateInfo **	out
	)		const

Status onnxruntime::KernelRegistry::TryFindKernel	(	const std::string &	op_name,
		const std::string &	domain,
		const int &	version,
		const std::unordered_map< std::string, MLDataType > &	type_constraints,
		ProviderType	exec_provider,
		const KernelCreateInfo **	out
	)		const

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