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

[CPU] Fix graph serialization, use ngraph serialization directly (#7261)

Browse Source
This commit is contained in:
Egor Duplensky 2021-09-03 11:05:20 +03:00 committed by GitHub
parent f33c03edd6
commit bf8113c056
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 11 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

13
inference-engine/src/mkldnn_plugin/mkldnn_graph_dumper.cpp
View File

@ -3,13 +3,17 @@
//
#include "mkldnn_graph_dumper.h"
#include "utils/debug_capabilities.h"
#include <ie_ngraph_utils.hpp>
#include "exec_graph_info.hpp"
#include "ie_common.h"
#include "mkldnn_debug.h"
#include <ngraph/variant.hpp>
#include "ngraph/ngraph.hpp"
#include "utils/debug_capabilities.h"
#include <ngraph/pass/manager.hpp>
#include <transformations/serialize.hpp>
#include <vector>
#include <string>
#include <memory>
@ -222,7 +226,12 @@ void serializeToXML(const MKLDNNGraph &graph, const std::string& path) {
if (path.empty())
return;
graph.dump().serialize(path);
std::string binPath;
ngraph::pass::Manager manager;
manager.register_pass<ngraph::pass::Serialize>(path,
binPath,
ngraph::pass::Serialize::Version::IR_V10);
manager.run_passes(graph.dump());
}
void serializeToCout(const MKLDNNGraph &graph) {