convert.h#

namespace ONNX_LIGHT_NAMESPACE
namespace version_conversion

Functions

ModelProto ConvertVersion(const ModelProto &mp_in, int target_version)#
class DefaultVersionConverter : public ONNX_LIGHT_NAMESPACE::version_conversion::BaseVersionConverter#
#include <convert.h>

Public Functions

inline DefaultVersionConverter()#
virtual ModelProto convert_version(const ModelProto &mp_in, const OpSetID &initial_version, const OpSetID &target_version) const override#

Converts a model from one opset version to another.

Parameters:

  • mp_in – Input model.

  • initial_version – Source opset identifier.

  • target_version – Target opset identifier.

Returns:

Converted model.

Private Functions

inline bool searchOpDomainMap(const std::unordered_map<std::string, std::map<int64_t, const OpSchema*>> &op_domain_map, int64_t curr_version, int64_t step) const#
inline void debug(const std::string &str) const#
inline void assertInVersionRange(int64_t version) const#
inline void assertDefaultDomain(const std::string &initial_domain, const std::string &target_domain) const#
void convert_graph(const std::shared_ptr<Graph> &g, const OpSetID &initial_version, const OpSetID &target_version) const#

Private Members

bool DEBUG = false#
std::pair<int, int> version_range#