From 3c269a059ec71137b05d8c0780e59feb8de0decc Mon Sep 17 00:00:00 2001
From: chujinjin <chujinjin52@huawei.com>
Date: Mon, 19 Oct 2020 11:16:43 +0800
Subject: [PATCH] opt multi fun

---
 mindspore/ops/composite/base.py | 3 +++
 1 file changed, 3 insertions(+)

diff --git a/mindspore/ops/composite/base.py b/mindspore/ops/composite/base.py
index d067fcc4ef..11e670909b 100644
--- a/mindspore/ops/composite/base.py
+++ b/mindspore/ops/composite/base.py
@@ -395,6 +395,9 @@ class MultitypeFuncGraph(MultitypeFuncGraph_):
                 sig.make_sig('args', sig.sig_rw.RW_READ, sig.sig_kind.KIND_VAR_POSITIONAL),))
 
     def __call__(self, *args):
+        if len(self.entries) == 1:
+            output = self.entries[0][1](*args)
+            return output
         types = tuple(map(mstype.get_py_obj_dtype, args))
         for sigs, fn in self.entries:
             if len(sigs) != len(types):