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

[MO] Fix axes in FusedBatchNorm -> MVN transformation (#7679) (#7913)

Browse Source
This commit is contained in:
Maxim Vafin
2021-10-12 16:47:29 +03:00
committed by GitHub
parent c323775f2c
commit 25c2d5c6c4
2 changed files with 9 additions and 9 deletions
Download Patch File Download Diff File Expand all files Collapse all files

14
model-optimizer/unit_tests/extensions/middle/FusedBatchNormTraining_test.py
View File

@@ -52,16 +52,16 @@ nodes_attributes = {
'reshape_to_orig': {'type': 'Reshape', 'value': None, 'kind': 'op', 'op': 'Reshape'},
'reshape_to_orig_data': {'value': None, 'shape': None, 'kind': 'data'},
'start': {'kind': 'op', 'op': 'Const'},
'start_data': {'value': None, 'shape': None, 'kind': 'data'},
'stop': {'kind': 'op', 'op': 'Const'},
'stop_data': {'value': None, 'shape': None, 'kind': 'data'},
'step': {'kind': 'op', 'op': 'Const'},
'step_data': {'value': None, 'shape': None, 'kind': 'data'},
'start': {'kind': 'op', 'op': 'Const', 'value': int64_array(1)},
'start_data': {'value': None, 'shape': None, 'kind': 'data', 'value': int64_array(1)},
'stop': {'kind': 'op', 'op': 'Const', 'value': int64_array(3)},
'stop_data': {'value': None, 'shape': None, 'kind': 'data', 'value': int64_array(3)},
'step': {'kind': 'op', 'op': 'Const', 'value': int64_array(1)},
'step_data': {'value': None, 'shape': None, 'kind': 'data', 'value': int64_array(1)},
'mvn_axes': {'kind': 'op', 'op': 'Range'},
'mvn_axes_data': {'value': None, 'shape': None, 'kind': 'data'},
'mvn': {'type': 'MVN', 'value': None, 'kind': 'op', 'op': 'MVN', 'eps': 1e-3},
'mvn': {'type': 'MVN', 'value': None, 'kind': 'op', 'op': 'MVN', 'eps': 1e-3, 'eps_mode': 'inside_sqrt'},
'mvn_data': {'value': None, 'shape': None, 'kind': 'data'},
'reshape_1': {'type': 'Reshape', 'value': None, 'kind': 'op', 'op': 'Reshape'},