IntenseWebs/openvino
3
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Translate all operation attributes values to lower case.

Browse Source
This commit is contained in:
jdanieck 2020-11-13 15:11:06 +01:00
parent c840afa383
commit d4c24175b3
1 changed files with 10 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

12
inference-engine/src/transformations/src/transformations/serialize.cpp
View File

@ -6,6 +6,7 @@
#include <fstream>
#include <unordered_map>
#include <unordered_set>
#include <algorithm>
#include <ngraph/variant.hpp>
#include "ngraph/ops.hpp"
@ -52,6 +53,13 @@ class XmlVisitor : public ngraph::AttributeVisitor {
return joinVec(adapter.get(), std::string(","));
}
std::string translate_attribute_value(std::string name) {
// IR format expects attribute values in lower case
std::transform(name.begin(), name.end(), name.begin(),
[](char c) { return std::tolower(c); });
return name;
}
public:
XmlVisitor(pugi::xml_node& data, std::string& node_type_name)
: m_data(data), m_node_type_name(node_type_name) {}
@ -75,8 +83,8 @@ public:
// it is a WA to not introduce dependency on plugin_api library
m_node_type_name = adapter.get();
} else {
m_data.append_attribute(name.c_str())
.set_value(adapter.get().c_str());
std::string value = translate_attribute_value(adapter.get());
m_data.append_attribute(name.c_str()).set_value(value.c_str());
}
}
void on_adapter(const std::string& name,