Searched refs:colocation_graph (Results 1 – 2 of 2) sorted by relevance
| /external/tensorflow/tensorflow/core/common_runtime/ |
| D | placer.cc | 68 ColocationGraph* colocation_graph, in AssignAndLog() argument
73 TF_RETURN_IF_ERROR(colocation_graph->LimitToAssignedDevice(*node)); in AssignAndLog()
113 ColocationGraph colocation_graph(graph_, devices_, default_device_, in Run() local
117 TF_RETURN_IF_ERROR(colocation_graph.Initialize()); in Run()
127 TF_RETURN_IF_ERROR(colocation_graph.LimitToAssignedDevice(*node)); in Run()
144 Status status = colocation_graph.GetDevicesForNode(node, &devices); in Run()
184 TF_RETURN_IF_ERROR(AssignAndLog(assigned_device, node, &colocation_graph, in Run()
192 Status status = colocation_graph.GetDevicesForNode(node, &devices); in Run()
224 TF_RETURN_IF_ERROR(AssignAndLog(assigned_device, node, &colocation_graph, in Run()
|
| /external/tensorflow/tensorflow/core/ |
| D | BUILD | 3092 "common_runtime/colocation_graph.cc",
3093 "common_runtime/colocation_graph.h",
|