#include <kernel_registry.h>

Public Types
using	TypeConstraintMap = InlinedHashMap< std::string, MLDataType >

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 Node &node, ProviderType exec_provider, const TypeConstraintMap &type_constraints, const KernelCreateInfo **out) const

Status	TryFindKernel (ProviderType exec_provider, std::string_view op_type, std::string_view domain, int version, const KernelRegistry::TypeConstraintMap &type_constraints, const KernelCreateInfo **out) const
	Find out whether a kernel is registered, without a node. This should be useful in graph optimizers, to check whether the node it is about to generate, is supported or not. More...

bool	IsEmpty () const

const KernelCreateMap &	GetKernelCreateMap () 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.

Member Typedef Documentation

using onnxruntime::KernelRegistry::TypeConstraintMap = InlinedHashMap<std::string, MLDataType>

Definition at line 39 of file kernel_registry.h.

Constructor & Destructor Documentation

onnxruntime::KernelRegistry::KernelRegistry ( )

default

Member Function Documentation

const KernelCreateMap& onnxruntime::KernelRegistry::GetKernelCreateMap ( ) const

inline

Definition at line 77 of file kernel_registry.h.

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

inlinestatic

Definition at line 66 of file kernel_registry.h.

bool onnxruntime::KernelRegistry::IsEmpty ( ) const

inline

Definition at line 74 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 Node &	node,
		ProviderType	exec_provider,
		const TypeConstraintMap &	type_constraints,
		const KernelCreateInfo **	out
	)		const

Status onnxruntime::KernelRegistry::TryFindKernel	(	ProviderType	exec_provider,
		std::string_view	op_type,
		std::string_view	domain,
		int	version,
		const KernelRegistry::TypeConstraintMap &	type_constraints,
		const KernelCreateInfo **	out
	)		const

Find out whether a kernel is registered, without a node. This should be useful in graph optimizers, to check whether the node it is about to generate, is supported or not.

Parameters

exec_provider
op_type
domain
version
type_constraints
out

Returns

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

onnxruntime/core/framework/kernel_registry.h

Public Types

Public Member Functions

Static Public Member Functions

Detailed Description

Member Typedef Documentation

Constructor & Destructor Documentation

Member Function Documentation