diff --git a/ngraph/core/src/op/broadcast.cpp b/ngraph/core/src/op/broadcast.cpp index 6df504f504e..c5992ca7ac9 100644 --- a/ngraph/core/src/op/broadcast.cpp +++ b/ngraph/core/src/op/broadcast.cpp @@ -205,7 +205,7 @@ shared_ptr 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 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 1621a3320a1..cb3f1a34092 100644 --- 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()} )