From 17c394c8fef09d37f944f3eb744fd8fa79cc913e Mon Sep 17 00:00:00 2001 From: Indira Salyahova Date: Wed, 26 Jan 2022 14:59:51 +0300 Subject: [PATCH] [POT] Update patterns (#9786) * Update special_patterns.py * add pattern in docs --- .../tools/pot/configs/hardware/IgnoredPatterns.md | 12 ++++++++++++ .../pot/openvino/tools/pot/graph/special_patterns.py | 9 +++++++++ 2 files changed, 21 insertions(+) diff --git a/tools/pot/openvino/tools/pot/configs/hardware/IgnoredPatterns.md b/tools/pot/openvino/tools/pot/configs/hardware/IgnoredPatterns.md index 5f4e96280ba..11800c70b6a 100755 --- a/tools/pot/openvino/tools/pot/configs/hardware/IgnoredPatterns.md +++ b/tools/pot/openvino/tools/pot/configs/hardware/IgnoredPatterns.md @@ -347,6 +347,18 @@ graph TB --- +**Name:** softmax
+**Pattern:**
+ +```mermaid +graph TB + exp(Exp) --> reduce(ReduceSum) + exp(Exp) --> divide(Divide) + reduce_const(Const) --> reduce(ReduceSum) + reduce(ReduceSum) --> divide(Divide) +``` +--- + **Name:** softmax_reshape_matmul
**Pattern:**
diff --git a/tools/pot/openvino/tools/pot/graph/special_patterns.py b/tools/pot/openvino/tools/pot/graph/special_patterns.py index 8cd423d4385..cad27e45a6b 100644 --- a/tools/pot/openvino/tools/pot/graph/special_patterns.py +++ b/tools/pot/openvino/tools/pot/graph/special_patterns.py @@ -144,6 +144,15 @@ def create_softmax_pattern(): return pattern.set_name('softmax').pattern +@registry_ignore_patterns('blocks') +def create_softmax_div_pattern(): + pattern = PatternBuilder() + exp_out = pattern.append_single_op('Exp', 'exp').get_last_node() + reduce_out = pattern.append_op_const('ReduceSum', 'reduce').get_last_node() + pattern.insert_single_op([exp_out, reduce_out], None, 'Divide', 'div') + return pattern.set_name('softmax_div').pattern + + @registry_ignore_patterns('blocks') def create_softmax_reshape_matmul_pattern(): pattern = PatternBuilder()