Moving rewrite_gemm to the end of the rewriters list

phager90 · phager90 · commit b97bed0bcb09 · 2020-08-21T09:30:15.000Z
diff --git a/tf2onnx/tfonnx.py b/tf2onnx/tfonnx.py
@@ -451,13 +451,13 @@ def compat_handler(ctx, node, **kwargs):
     # pre-processing graph rewrites
     # bi-directional re-writer should be placed after single directional re-writer
     rewriters = [rewrite_constant_fold, rewrite_quantize_and_dequantize, rewrite_transpose, rewrite_flatten,
-                 rewrite_gemm, rewrite_random_uniform, rewrite_random_uniform_fold_const,
+                 rewrite_random_uniform, rewrite_random_uniform_fold_const,
                  rewrite_random_normal, rewrite_dropout, rewrite_eye,
                  rewrite_leakyrelu, rewrite_thresholded_relu, rewrite_conv2d_with_pad,
                  rewrite_single_direction_lstm, rewrite_bi_direction_lstm,
                  rewrite_single_direction_gru, rewrite_bi_direction_gru,
                  rewrite_custom_rnn_cell, rewrite_generic_loop, rewrite_cond,
-                 rewrite_biasadd_with_conv2d,
+                 rewrite_biasadd_with_conv2d, rewrite_gemm
                  ]
 
     if custom_rewriter is not None:
