#2 Fix broadcasting issue in FQ ref implementation

vurusovs · Apr 14, 2023 · 259b4c1 · 259b4c1
1 parent 934793c
commit 259b4c1
Showing 1 changed file with 1 addition and 3 deletions.
diff --git a/src/core/reference/include/ngraph/runtime/reference/fake_quantize.hpp b/src/core/reference/include/ngraph/runtime/reference/fake_quantize.hpp
@@ -23,9 +23,7 @@ inline std::vector<size_t> calc_broadcast_index_offset(const std::vector<size_t>
                                                        const std::vector<size_t>& broadcast_shape) {
     std::vector<size_t> broadcast_offsets(broadcast_shape.size(), 0);
     for (int i = static_cast<int>(broadcast_shape.size()) - 2; i >= 0; --i) {
-        if (broadcast_shape[i] == 1) {
-            broadcast_offsets[i] = memory_offsets[i];
-        }
+        broadcast_offsets[i] = memory_offsets[i];
     }
     const auto not_one = [](size_t i) {
         return i != 1;