Searched refs:associated_functions (Results 1 – 3 of 3) sorted by relevance
147 auto associated_functions = GetAssociatedFunctions(*n, fld); in FunctionalizeControlFlowForNodeAssociatedFunctions() local148 if (!associated_functions.empty()) { in FunctionalizeControlFlowForNodeAssociatedFunctions()149 nodes_to_associated_functions.push_back({n, associated_functions}); in FunctionalizeControlFlowForNodeAssociatedFunctions()154 auto associated_functions = pair.second; in FunctionalizeControlFlowForNodeAssociatedFunctions() local155 for (auto& associated_function : associated_functions) { in FunctionalizeControlFlowForNodeAssociatedFunctions()165 associated_functions.size() == 1); in FunctionalizeControlFlowForNodeAssociatedFunctions()
396 const auto associated_functions = GetAssociatedFunctions(*n, library); in DetectHostTrainingLoop() local397 if (associated_functions.empty()) continue; in DetectHostTrainingLoop()400 associated_functions.begin(), in DetectHostTrainingLoop()401 associated_functions.end()); in DetectHostTrainingLoop()
1848 auto associated_functions = GetAssociatedFunctions( in InferXlaShardingFromNeighbors() local1850 for (auto& associated_function : associated_functions) { in InferXlaShardingFromNeighbors()