fix number error

openvinotoolkit · Feb 21, 2023 · 0d5e369 · 0d5e369
1 parent 8520d35
commit 0d5e369
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/src/frontends/paddle/src/op/flip.cpp b/src/frontends/paddle/src/op/flip.cpp
@@ -26,7 +26,7 @@ NamedOutputs flip(const NodeContext& node) {
             axis += dims;
         // Do nothing when dims of selected axis are 1.
         if (input_shape[axis] != 1) {
-            const auto split_axes = default_opset::Constant::create(element::i64, Shape{}, {input_shape[axis]});
+            const auto split_axes = default_opset::Constant::create(element::i64, Shape{}, {axis});
             auto split = std::make_shared<default_opset::Split>(temp, split_axes, axis);
             temp_split_out = split->outputs();
             // reverse the vector<Node> then concat