Searched refs:repl_node_ (Results 1 – 4 of 4) sorted by relevance
58 if (repl_node_.find(node) != repl_node_.end() || node->isa<ValueNode>()) { in CloneNode()83 repl_node_[node] = new_param; in CloneParameter()95 repl_node_[old_node] = new_node; in CloneCNode()107 repl_node_[node] = new_const; in CloneValueNode()119 repl_node_[node] = new_const; in CloneValueNode()132 if (repl_node_.count(old_node) == 0) { in CloneValueNodes()188 auto iter = repl_node_.find(old_return); in CloneFuncGraphValueNodes()189 if (iter == repl_node_.end()) { in CloneFuncGraphValueNodes()215 repl_node_[old_params[i]] = params[i]; in InlineCloneParameters()296 repl_node_[param] = node; in AddParameter()[all …]
62 std::unordered_map<AnfNodePtr, AnfNodePtr> *cloned_node() { return &repl_node_; } in cloned_node()69 std::unordered_map<AnfNodePtr, AnfNodePtr> repl_node_; variable
80 repl_node_ = nullptr; in ~FuncGraphSpecializer()97 std::unordered_map<AnfNodePtr, AnfNodePtr> *repl_node_; variable
127 repl_node_ = cloner_->cloned_node(); in FuncGraphSpecializer()142 MS_EXCEPTION_IF_NULL(specializer->repl_node_); in ReplicateDisconnectedNode()143 auto iter = specializer->repl_node_->find(node); in ReplicateDisconnectedNode()144 if (iter != specializer->repl_node_->end()) { in ReplicateDisconnectedNode()155 iter = specializer->repl_node_->find(node); in ReplicateDisconnectedNode()156 if (iter != specializer->repl_node_->end()) { in ReplicateDisconnectedNode()195 MS_EXCEPTION_IF_NULL(specializer->repl_node_); in GetReplicatedNode()196 auto iter = specializer->repl_node_->find(node); in GetReplicatedNode()197 if (iter != specializer->repl_node_->end()) { in GetReplicatedNode()