Searched refs:ColocationGraph (Results 1 – 4 of 4) sorted by relevance
| /external/tensorflow/tensorflow/core/common_runtime/ |
| D | colocation_graph.cc | 604 ColocationGraph::ColocationGraph(const Graph* graph, const FunctionStack& stack, in ColocationGraph() function in tensorflow::ColocationGraph
633 Status ColocationGraph::ColocateAllNodes() { in ColocateAllNodes()
685 Status ColocationGraph::ColocateResourceOrRefEdge(const Node* src, in ColocateResourceOrRefEdge()
715 Status ColocationGraph::ColocateResourceAndRefEdges( in ColocateResourceAndRefEdges()
789 Status ColocationGraph::AddHostOnlyDataTypesConstraints() { in AddHostOnlyDataTypesConstraints()
848 Status ColocationGraph::AddInspectionConstraints( in AddInspectionConstraints()
861 Status ColocationGraph::Initialize() { in Initialize()
944 Status ColocationGraph::ApplyIOColocationGroups( in ApplyIOColocationGroups()
1003 Status ColocationGraph::ColocateNodeToGroup( in ColocateNodeToGroup()
1038 Status ColocationGraph::ColocateNodes(const Node& x, const Node& y) { in ColocateNodes()
[all …]
|
| D | inspecting_placer.cc | 75 ColocationGraph* colocation_graph) in ColocationGraphToIOColocationGroups()
107 ColocationGraph* colocation_graph_;
148 ColocationGraph colocation_graph( in ComputeIOColocationGroups()
|
| D | colocation_graph.h | 212 class ColocationGraph {
217 ColocationGraph(const Graph* graph, const FunctionStack& stack,
381 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()
|