Home
last modified time | relevance | path

Searched refs:interference_map (Results 1 – 1 of 1) sorted by relevance

/external/tensorflow/tensorflow/compiler/xla/service/
Dbuffer_assignment.cc81 const std::vector<std::vector<int64>>& interference_map) { in ColorInterferenceGraph() argument
82 const int64 node_count = interference_map.size(); in ColorInterferenceGraph()
89 absl::c_sort(nodes, [&interference_map](const int64 i, const int64 j) { in ColorInterferenceGraph()
90 return interference_map[i].size() > interference_map[j].size(); in ColorInterferenceGraph()
98 for (int64 neighbor : interference_map[node]) { in ColorInterferenceGraph()
1496 std::vector<std::vector<int64>> interference_map( in MergeColocatedBufferSets() local
1501 interference_map[i].push_back(j); in MergeColocatedBufferSets()
1502 interference_map[j].push_back(i); in MergeColocatedBufferSets()
1509 auto assigned_colors = ColorInterferenceGraph(interference_map); in MergeColocatedBufferSets()