Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/compiler/xla/service/
Dbuffer_assignment.cc94 std::vector<int64> assigned_colors(node_count, kColorUnassigned); in ColorInterferenceGraph() local
99 int64 color = assigned_colors[neighbor]; in ColorInterferenceGraph()
113 assigned_colors[node] = color; in ColorInterferenceGraph()
115 return assigned_colors; in ColorInterferenceGraph()
1509 auto assigned_colors = ColorInterferenceGraph(interference_map); in MergeColocatedBufferSets() local
1512 CHECK(!assigned_colors.empty()); in MergeColocatedBufferSets()
1514 *std::max_element(assigned_colors.begin(), assigned_colors.end()) + 1; in MergeColocatedBufferSets()
1518 new_colocated_buffer_sets[assigned_colors[i]].insert(buffer_set.begin(), in MergeColocatedBufferSets()