use attribute name from spec

openvinotoolkit · Oct 10, 2024 · be046ac · be046ac
1 parent 9c8f261
commit be046ac
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 4 deletions.
diff --git a/src/core/include/openvino/op/squeeze.hpp b/src/core/include/openvino/op/squeeze.hpp
@@ -51,8 +51,8 @@ class OPENVINO_API Squeeze : public v0::Squeeze {
     ///
     /// \param data Input tensor with data
     /// \param axis The axis along which to squeeze the input tensor.
-    /// \param axis_skip_mode Shape inference result dynamic rank if selected axis has 1 in range of its dynamic
-    Squeeze(const Output<Node>& data, const Output<Node>& axes, const bool axis_skip_mode = false);
+    /// \param allow_axis_skip Shape inference result dynamic rank if selected axis has 1 in range of its dynamic
+    Squeeze(const Output<Node>& data, const Output<Node>& axes, const bool allow_axis_skip = false);
 
     std::shared_ptr<Node> clone_with_new_inputs(const OutputVector& new_args) const override;
 

diff --git a/src/core/src/op/squeeze.cpp b/src/core/src/op/squeeze.cpp
@@ -125,9 +125,9 @@ bool Squeeze::is_dynamic() const {
 }  // namespace v0
 
 namespace v15 {
-Squeeze::Squeeze(const Output<Node>& data, const Output<Node>& axes, const bool axis_skip_mode)
+Squeeze::Squeeze(const Output<Node>& data, const Output<Node>& axes, const bool allow_axis_skip)
     : v0::Squeeze({data, axes}),
-      m_allow_axis_skip{axis_skip_mode} {
+      m_allow_axis_skip{allow_axis_skip} {
     constructor_validate_and_infer_types();
 }