Searched refs:ColocationGraph (Results 1 – 5 of 5) sorted by relevance
| /external/tensorflow/tensorflow/core/common_runtime/ |
| D | colocation_graph.cc | 574 ColocationGraph::ColocationGraph(const Graph* graph, const FunctionStack& stack, in ColocationGraph() function in tensorflow::ColocationGraph
603 Status ColocationGraph::ColocateAllNodes() { in ColocateAllNodes()
655 Status ColocationGraph::ColocateResourceOrRefEdge(const Node* src, in ColocateResourceOrRefEdge()
679 Status ColocationGraph::ColocateResourceAndRefEdges( in ColocateResourceAndRefEdges()
729 Status ColocationGraph::AddInspectionConstraints( in AddInspectionConstraints()
742 Status ColocationGraph::Initialize() { in Initialize()
815 Status ColocationGraph::ApplyIOColocationGroups( in ApplyIOColocationGroups()
874 Status ColocationGraph::ColocateNodeToGroup( in ColocateNodeToGroup()
909 Status ColocationGraph::ColocateNodes(const Node& x, const Node& y) { in ColocateNodes()
918 Status ColocationGraph::ColocateNodes(const Node& x, int x_root, const Node& y, in ColocateNodes()
[all …]
|
| D | inspecting_placer.cc | 74 ColocationGraph* colocation_graph) in ColocationGraphToIOColocationGroups()
106 ColocationGraph* colocation_graph_;
147 ColocationGraph colocation_graph( in ComputeIOColocationGroups()
|
| D | colocation_graph.h | 208 class ColocationGraph {
213 ColocationGraph(const Graph* graph, const FunctionStack& stack,
369 TF_DISALLOW_COPY_AND_ASSIGN(ColocationGraph);
|
| D | placer.cc | 82 const ColocationGraph& colocation_graph) { in DumpColocationGraph()
121 ColocationGraph* colocation_graph, in AssignAndLog()
175 ColocationGraph colocation_graph(graph_, stack, flib_def_, devices_, in Run()
|
| /external/tensorflow/tensorflow/core/common_runtime/eager/ |
| D | context.cc | 186 selected = ColocationGraph::FilterSupportedDevices( in SelectDevice()
203 selected = ColocationGraph::FilterSupportedDevices( in SelectDevice()
216 selected = ColocationGraph::FilterSupportedDevices( in SelectDevice()
|