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()
121 ColocationGraph* colocation_graph, in AssignAndLog() argument
126 TF_RETURN_IF_ERROR(colocation_graph->LimitToAssignedDevice(*node)); in AssignAndLog()
175 ColocationGraph colocation_graph(graph_, stack, flib_def_, devices_, in Run() local
179 TF_RETURN_IF_ERROR(colocation_graph.Initialize()); in Run()
189 TF_RETURN_IF_ERROR(colocation_graph.LimitToAssignedDevice(*node)); in Run()
206 Status status = colocation_graph.GetDevicesForNode(node, &devices); in Run()
246 TF_RETURN_IF_ERROR(AssignAndLog(assigned_device, node, &colocation_graph, in Run()
254 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 | 222 "colocation_graph.h",
407 name = "colocation_graph",
409 "colocation_graph.cc",
413 "colocation_graph.h",
1164 ":colocation_graph",
|