#include <ONNXNodeConverterRegistry.h>

Public Types
using	ConverterFunc = void()(const onnx::NodeProto &onnx_node, ConverterContext context)

Public Member Functions
	NodeConverterRegistry ()=default

ConverterFunc	lookup (const std::string &optype, int64_t opset) const

void	registerConverter (const std::string &op_type, int64_t opset, ConverterFunc conv)

Static Public Member Functions
static NodeConverterRegistry &	getInstance ()

Detailed Description

Definition at line 60 of file ONNXNodeConverterRegistry.h.

Member Typedef Documentation

◆ ConverterFunc

using mir_onnx::NodeConverterRegistry::ConverterFunc = void (*)(const onnx::NodeProto &onnx_node, ConverterContext *context)

Definition at line 63 of file ONNXNodeConverterRegistry.h.

Constructor & Destructor Documentation

◆ NodeConverterRegistry()

mir_onnx::NodeConverterRegistry::NodeConverterRegistry ( )

default

Member Function Documentation

◆ getInstance()

NodeConverterRegistry & mir_onnx::NodeConverterRegistry::getInstance ( )

static

Definition at line 130 of file ONNXNodeConverterRegistry.cpp.

{
  static NodeConverterRegistry instance;
  return instance;
}

Referenced by mir_onnx::registerSupportedOps().

◆ lookup()

NodeConverterRegistry::ConverterFunc mir_onnx::NodeConverterRegistry::lookup	(	const std::string &	optype,
		int64_t	opset
	)		const

Definition at line 108 of file ONNXNodeConverterRegistry.cpp.

{
  auto it = _converter_map.find(optype);
  if (it == _converter_map.end())
  {
    return nullptr;
  }
 
  const VersionMap &conv_map = it->second;
 
  auto res = std::lower_bound(
    conv_map.crbegin(), conv_map.crend(), opset,
    [](const VersionMap::value_type &pair, int64_t opset) { return pair.first > opset; });
 
  if (res == conv_map.crend())
  {
    return nullptr;
  }
  return res->second;
}

◆ registerConverter()

void mir_onnx::NodeConverterRegistry::registerConverter	(	const std::string &	op_type,
		int64_t	opset,
		NodeConverterRegistry::ConverterFunc	conv
	)

Definition at line 136 of file ONNXNodeConverterRegistry.cpp.

{
  _converter_map[op_type].emplace(opset, conv);
}

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

compiler/mir/src/mir_onnx_importer/ONNXNodeConverterRegistry.h
compiler/mir/src/mir_onnx_importer/ONNXNodeConverterRegistry.cpp

Public Types

Public Member Functions

Static Public Member Functions