Searched defs:node_set (Results 1 – 13 of 13) sorted by relevance
36 const auto &node_set = iter->second; in UsedForOutputOnly() local
115 unsigned int node_set; in run() local
188 AnfNodeIndexSet node_set = iter->second; in IsCommunicationRelevantOp() local234 AnfNodeIndexSet node_set = iter->second; in SaveUserKernelTopoOrder() local
272 std::set<AnfNodePtr> node_set; in Run() local
140 AnfNodeIndexSet node_set = manager->node_users()[cnode]; in FindNextLayout() local580 auto &node_set = iter->second; in InitRefMap() local
120 auto node_set = node_user_map.at(cur_node->cast<AnfNodePtr>()); in ForwardSearchCNode() local
784 AnfNodeIndexSet node_set = manager->node_users()[node]; in StepSplitTensor() local1342 AnfNodeIndexSet node_set = manager->node_users()[node]; in FindParallelCareNode() local1686 int ObtainActualInputIdxForSupportedOps(const AnfNodeIndexSet &node_set) { in ObtainActualInputIdxForSupportedOps()1707 auto node_set = node_users_map.at(g_RefMap[parameter].first); in CoverSliceShape() local1727 auto node_set = node_users_map.at(res.first); in CoverSliceShape() local2050 AnfNodeIndexSet node_set = manager->node_users()[cnode]; in FindNextLayout() local3311 auto node_set = node_users_map.at(queue_node); in FindExpandDimsWIthGradScale() local
462 auto node_set = node_users_map.at(node); in RedistributionNextNode() local891 auto node_set = node_users_map.at(target_param); in BFSParallelCareNode() local946 auto node_set = node_users_map.at(queue_node); in GetChildCastNode() local2663 AnfNodeIndexSet node_set = manager->node_users()[anode]; in FindCNode() local
1195 AnfNodeIndexSet node_set = manager->node_users()[squeeze_cnode]; in HandleAdasumSlice() local1578 AnfNodeIndexSet node_set = manager->node_users()[node]; in FindParameterNextLayout() local
303 fn node_set<'q>( in node_set() function
494 auto node_set = node_users_map.at(node_ptr.first); local575 auto node_set = node_users_map[temp_node]; in HandleReceiveParam() local609 auto node_set = FindNextNode(temp_node, node_users_map); in AddVirtualAssignAdd() local
475 AnfNodeIndexSet node_set = manager->node_users()[cnode]; in FindReshapeNextNodeStraCosts() local
403 let mut node_set: HashSet<String> = HashSet::new(); in resolve_imports() localVariable