support BatchMatMulV2

zhijxu-MS · zhijxu-MS · commit a64c160f2431 · 2019-06-26T14:43:16.000+08:00
diff --git a/tf2onnx/onnx_opset/math.py b/tf2onnx/onnx_opset/math.py
@@ -271,7 +271,7 @@ def version_1(cls, ctx, node, **kwargs):
                                       name=op_name, shapes=shapes, dtypes=dtypes)
 
 
-@tf_op(["MatMul", "BatchMatMul"])
+@tf_op(["MatMul", "BatchMatMul", "BatchMatMulV2"])
 class MatMul:
     @classmethod
     def version_1(cls, ctx, node, **kwargs):
