Remove Matmul from broadcast op (#129)

microsoft · Jun 17, 2020 · c5c29c9 · c5c29c9
1 parent 3816b85
commit c5c29c9
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/onnxconverter_common/optimizer.py b/onnxconverter_common/optimizer.py
@@ -1075,7 +1075,7 @@ def find(node):
 
 
 _broadcast_types = {'Add', 'And', 'Div', 'Equal', 'Greater', 'GreaterOrEqual', 'Less', 'LessOrEqual',
-                    'MatMul', 'Max', 'Mean', 'Min', 'Mod', 'Mul', 'Or', 'Pow', 'PRelu', 'Sub', 'Sum',
+                    'Max', 'Mean', 'Min', 'Mod', 'Mul', 'Or', 'Pow', 'PRelu', 'Sub', 'Sum',
                     'Where', 'Xor'}
 _transpose_pass_type_set = {'Pad', 'Squeeze', 'Unsqueeze', 'Slice'}
 _transpose_pass_type_set.update(_broadcast_types)

diff --git a/tests/test_opt.py b/tests/test_opt.py
@@ -302,7 +302,7 @@ def test_merge_common(self):
     def test_onnx_models(self):
         model_names = ['mobile_segnet_no_opt.onnx', 'srgan_no_opt.onnx', 'test_model_0_no_opt.onnx',
                        'test_model_1_no_opt.onnx']
-        num_transpose_list = [2, 3, 11, 4]
+        num_transpose_list = [2, 3, 11, 5]
         dir_path = os.path.dirname(os.path.realpath(__file__))
         for idx_, model_name_ in enumerate(model_names):
             model_dir = dir_path + '/data/' + model_name_