onnxruntime::CustomRegistry Class Reference

#include <customregistry.h>

Public Member Functions
	CustomRegistry ()
common::Status	RegisterCustomKernel (KernelDefBuilder &kernel_def_builder, const KernelCreateFn &kernel_creator)
common::Status	RegisterCustomKernel (KernelCreateInfo &)
const std::shared_ptr < KernelRegistry > &	GetKernelRegistry ()
common::Status	RegisterOpSet (std::vector< ONNX_NAMESPACE::OpSchema > &schemas, const std::string &domain, int baseline_opset_version, int opset_version)
const std::shared_ptr < onnxruntime::OnnxRuntimeOpSchemaRegistry > &	GetOpschemaRegistry ()

Detailed Description

Represents a registry that contains both custom kernels and custom schemas.

Definition at line 21 of file customregistry.h.

Constructor & Destructor Documentation

onnxruntime::CustomRegistry::CustomRegistry ( )

inline

Definition at line 23 of file customregistry.h.

Member Function Documentation

const std::shared_ptr<KernelRegistry>& onnxruntime::CustomRegistry::GetKernelRegistry

(

)

const std::shared_ptr<onnxruntime::OnnxRuntimeOpSchemaRegistry>& onnxruntime::CustomRegistry::GetOpschemaRegistry

(

)

common::Status onnxruntime::CustomRegistry::RegisterCustomKernel	(	KernelDefBuilder &	kernel_def_builder,
		const KernelCreateFn &	kernel_creator
	)

Register a kernel definition together with kernel factory method to this session. If any conflict happened between registered kernel def and built-in kernel def, registered kernel will have higher priority. Call this before invoking Initialize().

Returns

OK if success.

common::Status onnxruntime::CustomRegistry::RegisterCustomKernel

(

KernelCreateInfo &

)

common::Status onnxruntime::CustomRegistry::RegisterOpSet	(	std::vector< ONNX_NAMESPACE::OpSchema > &	schemas,
		const std::string &	domain,
		int	baseline_opset_version,
		int	opset_version
	)

---

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

• onnxruntime/core/framework/customregistry.h