#include <ONNXNodeConverterRegistry.h>

Public Member Functions
	ModelContext (const onnx::ModelProto *model)

void	setDomainOpsetVersion (const std::string &domain, const int64_t opset_version)

int64_t	getDomainOpsetVersion (const std::string &domain) const

Detailed Description

Definition at line 30 of file ONNXNodeConverterRegistry.h.

Constructor & Destructor Documentation

◆ ModelContext()

mir_onnx::ModelContext::ModelContext ( const onnx::ModelProto * model )

explicit

Definition at line 37 of file ONNXNodeConverterRegistry.cpp.

{
  if (model == nullptr)
  {
    throw std::runtime_error{"Model should be imported before importer prepare"};
  }
 
  if (model->ir_version() > onnx::IR_VERSION)
  {
    throw std::runtime_error("IR version " + std::to_string(model->ir_version()) +
                             " is not supported yet.");
  }
 
  // Set Opset Version for each domain
  for (const auto &op_set : model->opset_import())
  {
    setDomainOpsetVersion(op_set.domain(), op_set.version());
  }
}

References setDomainOpsetVersion().

Member Function Documentation

◆ getDomainOpsetVersion()

int64_t mir_onnx::ModelContext::getDomainOpsetVersion ( const std::string & domain ) const

Definition at line 29 of file ONNXNodeConverterRegistry.cpp.

{
  auto iter = _domainToOpsetVersion.find(domain);
  if (iter == _domainToOpsetVersion.end())
    throw std::runtime_error("Didn't have domain " + domain + "!");
  return iter->second;
}

◆ setDomainOpsetVersion()

void mir_onnx::ModelContext::setDomainOpsetVersion	(	const std::string &	domain,
		const int64_t	opset_version
	)

Definition at line 24 of file ONNXNodeConverterRegistry.cpp.

{
  _domainToOpsetVersion.emplace(domain, opset_version);
}

Referenced by ModelContext().

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