Searched refs:colocation_graph (Results 1 – 3 of 3) sorted by relevance
| /external/tensorflow/tensorflow/core/common_runtime/ |
| D | placer.cc | 82 const ColocationGraph& colocation_graph) { in DumpColocationGraph() argument
87 colocation_graph.DebugString()); in DumpColocationGraph()
131 ColocationGraph* colocation_graph, in AssignAndLog() argument
136 TF_RETURN_IF_ERROR(colocation_graph->LimitToAssignedDevice(*node)); in AssignAndLog()
185 ColocationGraph colocation_graph(graph_, stack, flib_def_, devices_, in Run() local
189 TF_RETURN_IF_ERROR(colocation_graph.Initialize()); in Run()
199 TF_RETURN_IF_ERROR(colocation_graph.LimitToAssignedDevice(*node)); in Run()
216 Status status = colocation_graph.GetDevicesForNode(node, &devices); in Run()
258 TF_RETURN_IF_ERROR(AssignAndLog(assigned_device, node, &colocation_graph, in Run()
266 Status status = colocation_graph.GetDevicesForNode(node, &devices); in Run()
[all …]
|
| D | inspecting_placer.cc | 75 ColocationGraph* colocation_graph) in ColocationGraphToIOColocationGroups() argument
76 : colocation_graph_(colocation_graph), next_group_id_(0) {} in ColocationGraphToIOColocationGroups()
148 ColocationGraph colocation_graph( in ComputeIOColocationGroups() local
151 TF_RETURN_IF_ERROR(colocation_graph.Initialize()); in ComputeIOColocationGroups()
153 ColocationGraphToIOColocationGroups converter(&colocation_graph); in ComputeIOColocationGroups()
|
| D | BUILD | 255 "colocation_graph.h",
468 name = "colocation_graph",
470 "colocation_graph.cc",
474 "colocation_graph.h",
1371 ":colocation_graph",
|