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

[CPU] MatMulToFullyConnected transformation temporary disabled (#8364)

Browse Source
This commit is contained in:
Vladislav Golubev 2021-11-03 09:10:35 +03:00 committed by GitHub
parent e3f1813e5c
commit 983d55fba5
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 6 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
inference-engine/src/mkldnn_plugin/ngraph_transformations/convert_to_cpu_specific_opset.hpp
View File

@ -48,6 +48,12 @@ inline void ConvertToCPUSpecificOpset(std::shared_ptr<ngraph::Function> &nGraphF
}
manager.register_pass<ngraph::pass::ConstantFolding>();
manager.register_pass<ngraph::pass::ConvertPrecision>(precisions_array {{ ngraph::element::i64, ngraph::element::i32 }});
// TODO: remove after dynamic shapes support in FullyConnectedNode
manager.get_pass_config()->set_callback<ConvertMatMulToFC>([](const std::shared_ptr<const ngraph::Node>& node) -> bool {
return node->get_input_partial_shape(0).is_dynamic();
});
manager.run_passes(nGraphFunc);
}