Skip to content

Commit b65caba

Browse files
committed
fixed tests
1 parent 55e072f commit b65caba

2 files changed

Lines changed: 7 additions & 7 deletions

File tree

src/common/transformations/src/transformations/common_optimizations/nop_elimination.cpp

Lines changed: 4 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -517,10 +517,10 @@ EliminateSequentialFakeQuantize::EliminateSequentialFakeQuantize() {
517517
}
518518

519519
// Create merged FQ with FQ1's input range and FQ2's output range.
520-
auto fq1_in_low_const = v0::Constant::create(element::f32, Shape{}, {static_cast<float>(fq1_in_low)});
521-
auto fq1_in_high_const = v0::Constant::create(element::f32, Shape{}, {static_cast<float>(fq1_in_high)});
522-
auto fq2_out_low_const = v0::Constant::create(element::f32, Shape{}, {static_cast<float>(fq2_out_low)});
523-
auto fq2_out_high_const = v0::Constant::create(element::f32, Shape{}, {static_cast<float>(fq2_out_high)});
520+
auto fq1_in_low_const = v0::Constant::create(element::f32, Shape{1}, {static_cast<float>(fq1_in_low)});
521+
auto fq1_in_high_const = v0::Constant::create(element::f32, Shape{1}, {static_cast<float>(fq1_in_high)});
522+
auto fq2_out_low_const = v0::Constant::create(element::f32, Shape{1}, {static_cast<float>(fq2_out_low)});
523+
auto fq2_out_high_const = v0::Constant::create(element::f32, Shape{1}, {static_cast<float>(fq2_out_high)});
524524

525525
auto merged_fq = std::make_shared<v0::FakeQuantize>(fq1->input_value(0),
526526
fq1_in_low_const,

src/common/transformations/tests/common_optimizations/nop_elimination.cpp

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -197,9 +197,9 @@ TEST_P(MergedFakeQuantizeTests, CompareFunctions) {
197197
}
198198

199199
const std::vector<MergedFqParams> merged_fq_params = {
200-
{{-2.0f, 2.0f, -1.0f, 1.0f}, 256, {-1.0f, 1.0f, -0.5f, 0.5f}, 256, "merge_compressed_output"},
201-
{{-4.0f, 4.0f, -2.0f, 2.0f}, 256, {-2.0f, 2.0f, -1.0f, 1.0f}, 256, "merge_different_input_ranges"},
202-
{{-3.0f, 3.0f, -1.5f, 1.5f}, 512, {-1.5f, 1.5f, -0.75f, 0.75f}, 512, "merge_odd_levels"},
200+
{{-2.0f, 2.0f, -1.0f, 1.0f}, 256, {-1.0f, 1.0f, -1.0f, 0.0f}, 256, "merge_compressed_output"},
201+
{{-4.0f, 4.0f, -2.0f, 2.0f}, 256, {-2.0f, 2.0f, -2.0f, 0.0f}, 256, "merge_different_input_ranges"},
202+
{{-3.0f, 3.0f, -1.5f, 1.5f}, 512, {-1.5f, 1.5f, -1.5f, 0.0f}, 512, "merge_even_levels"},
203203
};
204204

205205
INSTANTIATE_TEST_SUITE_P(MergedFakeQuantize,

0 commit comments

Comments
 (0)