[LPT] MoveFQ fix (#9821)

This commit is contained in:
Vladislav Golubev authored and GitHub committed 2022-01-21 10:06:17 +03:00
1 parent 78db56d92d
commit a3e2dbe353
2 files changed
+83 -2

No files matched your search

@@ -1776,7 +1776,8 @@ std::vector<std::vector<std::shared_ptr<ngraph::opset1::Constant>>> NetworkHelpe
}
for (size_t i = 0; i < currConstants.size(); ++i) {
std::vector<std::shared_ptr<ngraph::opset1::Constant>> newConstant;
if (currConstants[i]->output(0).get_shape()[concat_axis] == 1) {
const auto const_shape = currConstants[i]->get_shape();
if (ov::shape_size(const_shape) == 1 || const_shape[concat_axis] == 1) {
newConstant.push_back(currConstants[i]);
newConstants[i] = newConstant;
continue;