Merge pull request #627 from zhijxu-MS/support_batchmatmulv2

Support batchmatmulv2, and fix tf2onnx logging problem

Author: zhijxu-MS

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):

tf2onnx/verbose_logging.py

@@ -42,7 +42,8 @@ def basicConfig(**kwargs):  # pylint: disable=invalid-name, function-redefined
     if "format" not in kwargs:
         level = kwargs.get("level", _logging.root.level)
         kwargs["format"] = _BASIC_LOG_FORMAT if level >= INFO else _VERBOSE_LOG_FORMAT
-
+    # config will make effect only when root.handlers is empty, so add the following statement to make sure it
+    _logging.root.handlers = []
     _logging.basicConfig(**kwargs)
     set_tf_verbosity(_logging.getLogger().getEffectiveLevel())