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

Update ShapeOf extender (#1406)

Browse Source
This commit is contained in:
Anton Chetverikov 2020-07-23 14:07:34 +03:00 committed by GitHub
parent 9440561fa4
commit cdd5605c61
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 2 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
model-optimizer/mo/utils/ir_reader/extenders/shape_of_extender.py
View File

@ -24,4 +24,5 @@ class ShapeOfExtender(Extender):
@staticmethod @staticmethod
def extend(op: Node): def extend(op: Node):
op['output_type'] = destination_type_to_np_data_type(op.output_type) if op.has_valid('output_type'):
op['output_type'] = destination_type_to_np_data_type(op.output_type)