onert::api::KernelBuilder Class Reference

Collaboration diagram for onert::api::KernelBuilder:

Public Member Functions
	KernelBuilder (CustomKernelRegistry *registry)
std::unique_ptr< exec::IFunction >	buildKernel (const std::string &id, backend::custom::CustomKernelConfigParams &&params) const override
Public Member Functions inherited from onert::backend::custom::IKernelBuilder
virtual	~IKernelBuilder ()=default

Detailed Description

Definition at line 24 of file CustomKernelRegistry.cc.

Constructor & Destructor Documentation

KernelBuilder()

onert::api::KernelBuilder::KernelBuilder ( CustomKernelRegistry *  registry )

inline

Definition at line 27 of file CustomKernelRegistry.cc.

: _registry(registry) {}

Member Function Documentation

buildKernel()

std::unique_ptr< exec::IFunction > onert::api::KernelBuilder::buildKernel ( const std::string &  id, backend::custom::CustomKernelConfigParams &&  params  ) const

inlineoverridevirtual

Implements onert::backend::custom::IKernelBuilder.

Definition at line 30 of file CustomKernelRegistry.cc.

  {
    auto kernel = _registry->buildKernelForOp(id);
    kernel->configure(std::move(params));
 
    return kernel;
  }

References onert::api::CustomKernelRegistry::buildKernelForOp().

---

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

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