IntenseWebs/openvino
3
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Broadcast visit_attributes fix (#2649)

Browse Source 
* Broadcast visit_attributes fix

* Broadcast: Python API adjustment

* Revert back deserialization change
This commit is contained in:
Evgenya Stepyreva 2020-10-15 21:39:42 +03:00 committed by GitHub
parent 7cead20209
commit 6ae28bdd62
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
2 changed files with 3 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
ngraph/core/src/op/broadcast.cpp
View File

@ -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) bool op::v3::Broadcast::visit_attributes(AttributeVisitor& visitor)
{ {
visitor.on_attribute("broadcast_spec", m_mode); visitor.on_attribute("mode", m_mode);
return true; 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) 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; return true;
} }

2
ngraph/python/src/ngraph/opset3/ops.py
View File

@ -97,7 +97,7 @@ def broadcast(
if broadcast_spec.upper() == "EXPLICIT": if broadcast_spec.upper() == "EXPLICIT":
inputs.append(as_node(axes_mapping)) inputs.append(as_node(axes_mapping))
return _get_node_factory_opset3().create( return _get_node_factory_opset3().create(
"Broadcast", inputs, {"broadcast_spec": broadcast_spec.upper()} "Broadcast", inputs, {"mode": broadcast_spec.upper()}
) )