Broadcast visit_attributes fix (#2649)

* Broadcast visit_attributes fix

* Broadcast: Python API adjustment

* Revert back deserialization change

diff --git a/ngraph/core/src/op/broadcast.cpp b/ngraph/core/src/op/broadcast.cpp
index 6df504f..c5992ca 100644 (file)
--- a/ngraph/core/src/op/broadcast.cpp
+++ b/ngraph/core/src/op/broadcast.cpp
@@ -205,7 +205,7 @@ shared_ptr<Node> op::v3::Broadcast::clone_with_new_inputs(const OutputVector& ne
 
 bool op::v3::Broadcast::visit_attributes(AttributeVisitor& visitor)
 {
-    visitor.on_attribute("broadcast_spec", m_mode);
+    visitor.on_attribute("mode", m_mode);
     return true;
 }
 
@@ -286,7 +286,7 @@ shared_ptr<Node> op::v1::Broadcast::clone_with_new_inputs(const OutputVector& ne
 
 bool op::v1::Broadcast::visit_attributes(AttributeVisitor& visitor)
 {
-    visitor.on_attribute("broadcast_spec", m_broadcast_spec);
+    visitor.on_attribute("mode", m_broadcast_spec);
     return true;
 }
 

diff --git a/ngraph/python/src/ngraph/opset3/ops.py b/ngraph/python/src/ngraph/opset3/ops.py
index 1621a33..cb3f1a3 100644 (file)
--- a/ngraph/python/src/ngraph/opset3/ops.py
+++ b/ngraph/python/src/ngraph/opset3/ops.py
@@ -97,7 +97,7 @@ def broadcast(
     if broadcast_spec.upper() == "EXPLICIT":
         inputs.append(as_node(axes_mapping))
     return _get_node_factory_opset3().create(
-        "Broadcast", inputs, {"broadcast_spec": broadcast_spec.upper()}
+        "Broadcast", inputs, {"mode": broadcast_spec.upper()}
     )