Searched refs:device_to_index (Results 1 – 1 of 1) sorted by relevance
1764 absl::flat_hash_map<int64, std::vector<int64>> device_to_index; in FindMatchingPartitionedDimsForGrouping() local1767 device_to_index[device] = in FindMatchingPartitionedDimsForGrouping()1773 if (device_to_index[device_groups[0][0]][i] == in FindMatchingPartitionedDimsForGrouping()1774 device_to_index[device_groups[0][1]][i]) { in FindMatchingPartitionedDimsForGrouping()1785 return device_to_index[group[i]][dim] != in FindMatchingPartitionedDimsForGrouping()1786 device_to_index[group[0]][dim]; in FindMatchingPartitionedDimsForGrouping()