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

soft_get('op')

Browse Source
This commit is contained in:
Pavel Esir 2021-06-02 10:19:59 +03:00
parent 5f697c8539
commit cadfe18fa5
1 changed files with 2 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
model-optimizer/extensions/ops/elementwise.py
View File

@ -24,11 +24,11 @@ def override_data_type_of_constant(node: Node):
in_node_0 = node.in_port(0).get_source().node
in_node_1 = node.in_port(1).get_source().node
if in_node_0.op != 'Const' and in_node_1.op != 'Const':
if in_node_0.soft_get('op') != 'Const' and in_node_1.op != 'Const':
raise Exception("Elementwise operation '{}' has inputs of different data types: '{}' and '{}' "
'that cannot be aligned'.format(node.soft_get('name'), in_type_0, in_type_1))
if in_node_0.op == 'Const':
if in_node_0.soft_get('op') == 'Const':
node_to_convert, src_type, dst_type = in_node_0, in_type_0, in_type_1
else:
node_to_convert, src_type, dst_type = in_node_1, in_type_1, in_type_0