diff --git a/tools/pot/openvino/tools/pot/graph/passes.py b/tools/pot/openvino/tools/pot/graph/passes.py index d6bbc05cb6e..ff5adfbf546 100644 --- a/tools/pot/openvino/tools/pot/graph/passes.py +++ b/tools/pot/openvino/tools/pot/graph/passes.py @@ -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 diff --git a/tools/pot/tests/data/models/split_concat_example/split_concat_example.json b/tools/pot/tests/data/models/split_concat_example/split_concat_example.json new file mode 100644 index 00000000000..89701729bb6 --- /dev/null +++ b/tools/pot/tests/data/models/split_concat_example/split_concat_example.json @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:a92b39ec5936e038627cae4e3a080322aba4785f181fd3493d83d56014cc1c0c +size 172 diff --git a/tools/pot/tests/data/models/split_concat_example/split_concat_example.onnx b/tools/pot/tests/data/models/split_concat_example/split_concat_example.onnx new file mode 100644 index 00000000000..a71b2bd6813 --- /dev/null +++ b/tools/pot/tests/data/models/split_concat_example/split_concat_example.onnx @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:6e6b221ddce1eb1536ba7e5c3112e137dde45181e314ab436da5ff4fb236c9b6 +size 449 diff --git a/tools/pot/tests/data/reference_models/split_concat_example_pytorch.xml b/tools/pot/tests/data/reference_models/split_concat_example_pytorch.xml new file mode 100644 index 00000000000..075502bf529 --- /dev/null +++ b/tools/pot/tests/data/reference_models/split_concat_example_pytorch.xml @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:46ed853f0550eb2be4f8c48257c070aa98f8f811de7764e78074130f89f96f6b +size 8751 diff --git a/tools/pot/tests/test_graph.py b/tools/pot/tests/test_graph.py index 097ab2d0541..5f4380a2e13 100755 --- a/tools/pot/tests/test_graph.py +++ b/tools/pot/tests/test_graph.py @@ -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'), ]