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

Fix bug with unknown node type (#13406)

Browse Source
This commit is contained in:
Liubov Talamanova
2022-10-11 16:05:54 +01:00
committed by GitHub
parent abd58eef5c
commit 968bd6c96b
5 changed files with 12 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
tools/pot/openvino/tools/pot/graph/passes.py
View File

@@ -876,6 +876,8 @@ def insert_fake_quantize(graph, node, ports=None, names=None, fq_types=None, hw_
node_type = get_hardware_config_operation_type(node, list(hw_config.keys()))
if hw_config[node_type]:
fq_configs = hw_config[node_type][fq_group]
else:
node_type = None
if node_type in input_priority_types:
fq_config_priority = 2

3
tools/pot/tests/data/models/split_concat_example/split_concat_example.json Normal file
View File

@@ -0,0 +1,3 @@
version https://git-lfs.github.com/spec/v1
oid sha256:a92b39ec5936e038627cae4e3a080322aba4785f181fd3493d83d56014cc1c0c
size 172

3
tools/pot/tests/data/models/split_concat_example/split_concat_example.onnx Normal file
View File

@@ -0,0 +1,3 @@
version https://git-lfs.github.com/spec/v1
oid sha256:6e6b221ddce1eb1536ba7e5c3112e137dde45181e314ab436da5ff4fb236c9b6
size 449

3
tools/pot/tests/data/reference_models/split_concat_example_pytorch.xml Normal file
View File

@@ -0,0 +1,3 @@
version https://git-lfs.github.com/spec/v1
oid sha256:46ed853f0550eb2be4f8c48257c070aa98f8f811de7764e78074130f89f96f6b
size 8751

1
tools/pot/tests/test_graph.py
View File

@@ -30,6 +30,7 @@ TEST_MODELS = [
('resnet_example', 'pytorch', 'CPU_SPR'),
#('tensor_iterator_example', 'tf', 'ANY'),
('softsign_example', 'tf', 'GNA'),
('split_concat_example', 'pytorch', 'ANY'),
]