Searched refs:unmodified_output_to_input_map (Results 1 – 1 of 1) sorted by relevance
492 absl::flat_hash_map<int64_t, int64_t> unmodified_output_to_input_map; in MatchReduceScatter() local494 unmodified_output_to_input_map.insert({io_pair.second, io_pair.first}); in MatchReduceScatter()499 return unmodified_output_to_input_map.count(out_dim) != 0; in MatchReduceScatter()508 spec.split_dim = unmodified_output_to_input_map.at(spec.split_dim); in MatchReduceScatter()510 split_dim = unmodified_output_to_input_map.at(split_dim); in MatchReduceScatter()