Searched refs:reachable_functions (Results 1 – 4 of 4) sorted by relevance
44 ModuleOp module, SmallPtrSetImpl<Operation*>& reachable_functions) { in PopulateClusterReachableFunctions() argument65 if (reachable_functions.insert(func).second) in PopulateClusterReachableFunctions()77 if (reachable_functions.insert(callee).second) in PopulateClusterReachableFunctions()115 SmallPtrSet<Operation*, 16> reachable_functions; in ApplyPatternsInClusterAndReachableFunctions() local116 PopulateClusterReachableFunctions(module, reachable_functions); in ApplyPatternsInClusterAndReachableFunctions()119 for (Operation* op : reachable_functions) { in ApplyPatternsInClusterAndReachableFunctions()134 if (reachable_functions.contains(func)) continue; in ApplyPatternsInClusterAndReachableFunctions()
371 SmallPtrSet<Operation*, 16> reachable_functions; in FindTPUPartitionedCallReachableFunctions() local375 if (reachable_functions.insert(caller).second) { in FindTPUPartitionedCallReachableFunctions()381 return reachable_functions; in FindTPUPartitionedCallReachableFunctions()
972 FunctionLibraryDefinition reachable_functions = in PruneUnreachableFunctionsFromGraph() local975 if (!reachable_functions.Find(func_name)) { in PruneUnreachableFunctionsFromGraph()
4775 FunctionLibraryDefinition reachable_functions = in RewriteTPUReplicateNode() local4780 FingerprintFunctionLibrary(reachable_functions, &library_fingerprint)); in RewriteTPUReplicateNode()4782 << absl::StrJoin(reachable_functions.ListFunctionNames(), ", "); in RewriteTPUReplicateNode()4848 ContainsResourceWriteOp(*graph, reachable_functions); in RewriteTPUReplicateNode()