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

Fix serialization dynamic shapes (#3475)

Browse Source 
* Align EpsMode attribute to specification.

* Change dynamic shape resolving in serialization.
This commit is contained in:
Jozef Daniecki 2020-12-07 04:55:10 +01:00 committed by GitHub
parent 8581a0730d
commit fc049fc6ce
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
2 changed files with 3 additions and 5 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

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

@ -303,15 +303,13 @@ bool is_exec_graph(const ngraph::Function& f) {
} }
bool resolve_dynamic_shapes(const ngraph::Function& f) { bool resolve_dynamic_shapes(const ngraph::Function& f) {
const auto & f_results = f.get_results(); const auto & f_ops = f.get_ordered_ops();
if (std::all_of(f_results.begin(), f_results.end(), if (std::all_of(f_ops.begin(), f_ops.end(),
[](std::shared_ptr<Node> results) { return !results->is_dynamic(); })) { [](std::shared_ptr<Node> results) { return !results->is_dynamic(); })) {
return false; return false;
} }
auto f_clone = ngraph::clone_function(f); auto f_clone = ngraph::clone_function(f);
const auto & f_ops = f.get_ordered_ops();
const auto & f_clone_ops = f_clone->get_ordered_ops(); const auto & f_clone_ops = f_clone->get_ordered_ops();
NGRAPH_CHECK(f_ops.size() == f_clone_ops.size(), "Unexpected get_ordered_ops method behaviour"); NGRAPH_CHECK(f_ops.size() == f_clone_ops.size(), "Unexpected get_ordered_ops method behaviour");

2
ngraph/core/src/op/util/attr_types.cpp
View File

@ -118,7 +118,7 @@ namespace ngraph
NGRAPH_API EnumNames<op::EpsMode>& EnumNames<op::EpsMode>::get() NGRAPH_API EnumNames<op::EpsMode>& EnumNames<op::EpsMode>::get()
{ {
static auto enum_names = EnumNames<op::EpsMode>( static auto enum_names = EnumNames<op::EpsMode>(
"op::EpsMode", {{"ADD", op::EpsMode::ADD}, {"MAX", op::EpsMode::MAX}}); "op::EpsMode", {{"add", op::EpsMode::ADD}, {"max", op::EpsMode::MAX}});
return enum_names; return enum_names;
} }