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

Fix mode attribute value in DepthToSpace ONNX operation extractor (#3466)

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

4
model-optimizer/extensions/front/onnx/depth_to_space_ext.py
View File

@ -34,9 +34,9 @@ class DepthToSpaceFrontExtractor(FrontExtractorOp):
onnx_mode = onnx_attr(node, 'mode', 's', default=b'DCR').decode() onnx_mode = onnx_attr(node, 'mode', 's', default=b'DCR').decode()
assert onnx_mode in ['DCR', 'CRD'], 'Unrecognized mode provided for DepthToSpace node {}'.format(node_name) assert onnx_mode in ['DCR', 'CRD'], 'Unrecognized mode provided for DepthToSpace node {}'.format(node_name)
if onnx_mode == 'DCR': if onnx_mode == 'DCR':
mode = 'depth_first'
else:
mode = 'blocks_first' mode = 'blocks_first'
else:
mode = 'depth_first'
DepthToSpaceOp.update_node_stat(node, {'block_size': block_size, 'mode': mode}) DepthToSpaceOp.update_node_stat(node, {'block_size': block_size, 'mode': mode})
return cls.enabled return cls.enabled