From 786c5074da4d03260ff70a0b8e7b346a66ad23fe Mon Sep 17 00:00:00 2001
From: Kang <wuyongkang7@huawei.com>
Date: Wed, 10 Jun 2020 19:23:32 +0800
Subject: [PATCH] Optimized IsTrivial function

---
 mindspore/ccsrc/optimizer/irpass/inline.h | 6 +-----
 1 file changed, 1 insertion(+), 5 deletions(-)

diff --git a/mindspore/ccsrc/optimizer/irpass/inline.h b/mindspore/ccsrc/optimizer/irpass/inline.h
index 4e79d81017..854b568453 100644
--- a/mindspore/ccsrc/optimizer/irpass/inline.h
+++ b/mindspore/ccsrc/optimizer/irpass/inline.h
@@ -71,11 +71,7 @@ class ReplaceApplicator : public AnfVisitor {
 using CriterionFuncType = std::function<bool(FuncGraphPtr, AnfNodePtr)>;
 
 bool IsTrivial(const FuncGraphPtr &fg, AnfNodePtr) {
-  auto &s = fg->nodes();
-  int n_cnode = std::count_if(s.begin(), s.end(), [](const AnfNodePtr &n) {
-    MS_EXCEPTION_IF_NULL(n);
-    return n->isa<CNode>();
-  });
+  auto n_cnode = fg->nodes().size() - fg->parameters().size();
   // There is at least one CNode(return, other_node).
   return n_cnode <= 2;
 }