diff --git a/docs/MO_DG/prepare_model/Supported_Frameworks_Layers.md b/docs/MO_DG/prepare_model/Supported_Frameworks_Layers.md index 30173058227..42468e1d82f 100644 --- a/docs/MO_DG/prepare_model/Supported_Frameworks_Layers.md +++ b/docs/MO_DG/prepare_model/Supported_Frameworks_Layers.md @@ -292,6 +292,7 @@ Standard TensorFlow\* operations: | Pow | No | | Rsqrt | No | | Select | No | +| SelectV2 | No | | Shape | No | | Sigmoid | No | | Sin | No | diff --git a/model-optimizer/extensions/front/tf/select_ext.py b/model-optimizer/extensions/front/tf/select_ext.py index 4a2436a8f21..bcd15115795 100644 --- a/model-optimizer/extensions/front/tf/select_ext.py +++ b/model-optimizer/extensions/front/tf/select_ext.py @@ -14,3 +14,13 @@ class SelectExtractor(FrontExtractorOp): def extract(cls, node: Node): Select.update_node_stat(node, {'format': 'tf',}) return cls.enabled + + +class SelectV2Extractor(FrontExtractorOp): + op = 'SelectV2' + enabled = True + + @classmethod + def extract(cls, node: Node): + Select.update_node_stat(node, {'format': 'tf'}) + return cls.enabled