Searched refs:per_tuple_element_gtes (Results 1 – 1 of 1) sorted by relevance
205 std::vector<std::vector<HloInstruction*>> per_tuple_element_gtes( in HandleAllReduce() local211 per_tuple_element_gtes[user->tuple_index()].push_back(user); in HandleAllReduce()217 auto all_gte_users_are_bf16_convert = [&per_tuple_element_gtes, i]() { in HandleAllReduce()218 for (auto gte : per_tuple_element_gtes[i]) { in HandleAllReduce()231 for (auto gte : per_tuple_element_gtes[i]) { in HandleAllReduce()