[GPU] Fixed axes normalization for squeeze (#26082)

### Tickets: - *[149721](https://jira.devtools.intel.com/browse/CVS-149721)*
openvinotoolkit · Aug 16, 2024 · 09f1f91 · 09f1f91
1 parent 8b82aae
commit 09f1f91
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/src/plugins/intel_gpu/src/plugin/ops/reshape.cpp b/src/plugins/intel_gpu/src/plugin/ops/reshape.cpp
@@ -31,8 +31,10 @@ static void CreateCommonReshapeOp(ProgramBuilder& p, const std::shared_ptr<ov::N
         std::vector<int64_t> output_pattern = {};
         if (second_const_input != nullptr) {
             output_pattern = second_const_input->cast_vector<int64_t>();
-            if (mode == cldnn::reshape::reshape_mode::unsqueeze || mode == cldnn::reshape::reshape_mode::squeeze) {
+            if (mode == cldnn::reshape::reshape_mode::unsqueeze) {
                 ov::util::try_normalize_axes(output_pattern, op->get_output_partial_shape(0).rank(), *op);
+            } else if (mode == cldnn::reshape::reshape_mode::squeeze) {
+                ov::util::try_normalize_axes(output_pattern, op->get_input_partial_shape(0).rank(), *op);
             }
         }