!7443 opt multi function match machanism

Merge pull request !7443 from chujinjin/opt_multi_fun
pull/7443/MERGE
mindspore-ci-bot 5 years ago committed by Gitee
commit 9db4900837

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

Loading…
Cancel
Save