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

add selectv2 extractor mapped selectv2 to select (#7847)

Browse Source 
* add selectv2 extractor mapped selectv2 to select (broadcast already implemented in select transformations/implementations)

* added SelectV2 to supported layers list
This commit is contained in:
Svetlana Dolinina 2021-10-13 14:53:15 +03:00 committed by GitHub
parent 9e79f997e4
commit fd97a62263
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
2 changed files with 11 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

1
docs/MO_DG/prepare_model/Supported_Frameworks_Layers.md
View File

@ -292,6 +292,7 @@ Standard TensorFlow\* operations:
| Pow | No | | Pow | No |
| Rsqrt | No | | Rsqrt | No |
| Select | No | | Select | No |
| SelectV2 | No |
| Shape | No | | Shape | No |
| Sigmoid | No | | Sigmoid | No |
| Sin | No | | Sin | No |

10
model-optimizer/extensions/front/tf/select_ext.py
View File

@ -14,3 +14,13 @@ class SelectExtractor(FrontExtractorOp):
def extract(cls, node: Node): def extract(cls, node: Node):
Select.update_node_stat(node, {'format': 'tf',}) Select.update_node_stat(node, {'format': 'tf',})
return cls.enabled 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