Registry of kernel builders. More...

Public Types
using	KernelBuilderFunc = std::unique_ptr< Kernel >(const luci::CircleNode *, KernelBuilderHelper &)

Public Member Functions
	KernelBuilderRegistry ()

KernelBuilderFunc *	get_kernel_builder_func (luci::CircleOpcode opcode) const

Detailed Description

Registry of kernel builders.

This class contains mapping from Opcodes to kernel builder functions

Definition at line 74 of file KernelBuilder.cpp.

Member Typedef Documentation

◆ KernelBuilderFunc

using luci_interpreter::KernelBuilderRegistry::KernelBuilderFunc = std::unique_ptr<Kernel>(const luci::CircleNode *, KernelBuilderHelper &)

Definition at line 77 of file KernelBuilder.cpp.

Constructor & Destructor Documentation

◆ KernelBuilderRegistry()

luci_interpreter::KernelBuilderRegistry::KernelBuilderRegistry ( )

inline

Definition at line 80 of file KernelBuilder.cpp.

                          : _operator_builders(size_t(BuilderId::Size), nullptr)
  {
#define REGISTER_KERNEL(name) \
  register_kernel_builder(BuilderId::Circle##name, build_kernel_Circle##name);
 
#include "KernelsToBuild.lst"
 
#undef REGISTER_KERNEL
  }

Member Function Documentation

◆ get_kernel_builder_func()

KernelBuilderFunc * luci_interpreter::KernelBuilderRegistry::get_kernel_builder_func ( luci::CircleOpcode opcode ) const

inline

Definition at line 90 of file KernelBuilder.cpp.

  {
    return _operator_builders.at(size_t(opcode));
  }

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

compiler/luci-interpreter/src/loader/KernelBuilder.cpp

Public Types