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

Fix backward compatibility of MVN in MO (#4289)

Browse Source
This commit is contained in:
Maxim Vafin 2021-02-11 22:36:31 +03:00 committed by GitHub
parent 3db73ec552
commit d91b9fd8da
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 4 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
model-optimizer/extensions/ops/mvn.py
View File

@ -60,11 +60,13 @@ class MVN(Op):
name = node.soft_get('name', node.id) name = node.soft_get('name', node.id)
assert node.eps is not None, 'MVN required attribute `eps` unspecified for node {}'.format(name) assert node.eps is not None, 'MVN required attribute `eps` unspecified for node {}'.format(name)
assert node.eps_mode is not None, 'MVN required attribute `eps_mode` unspecified for node {}'.format(name)
assert node.normalize_variance is not None, \ assert node.normalize_variance is not None, \
'MVN required attribute `normalize_variance` unspecified for node {}'.format(name) 'MVN required attribute `normalize_variance` unspecified for node {}'.format(name)
PermuteInputs().set_input_permutation(node.in_node(1), node, 'input:0', 'axis') if node.version == 'opset6':
assert node.eps_mode is not None, 'MVN required attribute `eps_mode` unspecified for node {}'.format(name)
PermuteInputs().set_input_permutation(node.in_node(1), node, 'input:0', 'axis')
copy_shape_infer(node) copy_shape_infer(node)