onert::api::CustomKernelRegistry Class Reference

#include <CustomKernelRegistry.h>

Public Member Functions
void	registerKernel (const std::string &id, nnfw_custom_eval evalFunction)
std::shared_ptr< backend::custom::IKernelBuilder >	getBuilder ()
std::unique_ptr< CustomKernel >	buildKernelForOp (const std::string &id)

Detailed Description

Definition at line 33 of file CustomKernelRegistry.h.

Member Function Documentation

buildKernelForOp()

std::unique_ptr< CustomKernel > onert::api::CustomKernelRegistry::buildKernelForOp

(

const std::string &

id

)

Definition at line 53 of file CustomKernelRegistry.cc.

{
  auto it = _storage.find(id);
  if (it == _storage.end())
  {
    throw std::runtime_error("Unable to find associated kernel for op");
  }
 
  return std::make_unique<CustomKernel>(it->second);
}

Referenced by onert::api::KernelBuilder::buildKernel().

getBuilder()

std::shared_ptr< backend::custom::IKernelBuilder > onert::api::CustomKernelRegistry::getBuilder

(

)

Definition at line 48 of file CustomKernelRegistry.cc.

{
  return std::make_unique<KernelBuilder>(this);
}

registerKernel()

void onert::api::CustomKernelRegistry::registerKernel	(	const std::string &	id,
		nnfw_custom_eval	evalFunction
	)

Definition at line 43 of file CustomKernelRegistry.cc.

{
  _storage.emplace(id, evalFunction);
}

---

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

• runtime/onert/api/nnfw/src/CustomKernelRegistry.h
• runtime/onert/api/nnfw/src/CustomKernelRegistry.cc