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

Set fullname instead of name in fq_configuration (#9281)

Browse Source
This commit is contained in:
Liubov Talamanova 2021-12-17 18:46:57 +03:00 committed by GitHub
parent ba4802554e
commit ad10edbd2e
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 3 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
tools/pot/openvino/tools/pot/algorithms/quantization/fake_quantize_configuration.py
View File

@ -398,11 +398,11 @@ def _fake_quantize_to_types(model, hardware_config):
if not _is_quantizable(child):
queue.append(child)
elif child.type not in descendants:
descendants.append((child.name,
descendants.append((child.fullname,
get_hardware_config_operation_type(child, available_types)))
if current.type == 'Split' \
and child.type == 'Concat' \
and len({child_.name for child_ in children}) == 1:
and len({child_.fullname for child_ in children}) == 1:
break
return descendants
@ -414,7 +414,7 @@ def _fake_quantize_to_types(model, hardware_config):
available_types = [layer['type'] for layer in hardware_config]
for fq in get_nodes_by_type(model, ['FakeQuantize']):
node_input = get_node_input(fq, 0)
out[fq.name] = (_get_node_valuable_descendant(fq), node_input.type == 'Const')
out[fq.fullname] = (_get_node_valuable_descendant(fq), node_input.type == 'Const')
return out