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

[LPT] Fixed an incorrect condition & added test to MoveFakeQuantize transformation (#10009)

Browse Source 
* fixed an incorrect condition & added test

* fixed an incorrect condition & added test
This commit is contained in:
Nikita Demashov authored and GitHub committed 2022-02-07 12:32:49 +03:00
1 parent b365e67561
commit 74fa60cf86
2 files changed
+65 -1
Download Patch File Download Diff File Expand all files Collapse all files

No files matched your search

src/common/low_precision_transformations/src/move_fake_quantize.cpp
+1 -1
View File
@@ -86,7 +86,7 @@ bool MoveFakeQuantize::transform(TransformationContext& context, ngraph::pattern
const auto concat_axis = concat_node->get_concatenation_axis();
for (size_t i = 0; i < 4; i++) {
curr_constants[i] = as_type_ptr<opset1::Constant>(fq->get_input_node_shared_ptr(i + 1));
if (!multi_chanels && curr_constants[i]->get_shape().size() > (concat_axis + 1ul) && curr_constants[i]->get_shape()[concat_axis] != 1) {
if (!multi_chanels && curr_constants[i]->get_shape().size() > concat_axis && curr_constants[i]->get_shape()[concat_axis] != 1) {
multi_chanels = true;
}
}