Searched refs:ColocationGraph (Results 1 – 4 of 4) sorted by relevance
| /aosp_15_r20/external/tensorflow/tensorflow/core/common_runtime/ |
| H A D | colocation_graph.cc | 627 ColocationGraph::ColocationGraph(const Graph* graph, const FunctionStack& stack, in ColocationGraph() function in tensorflow::ColocationGraph
655 Status ColocationGraph::ColocateAllNodes() { in ColocateAllNodes()
707 Status ColocationGraph::ColocateResourceOrRefEdge(const Node* src, in ColocateResourceOrRefEdge()
737 Status ColocationGraph::ColocateResourceAndRefEdges( in ColocateResourceAndRefEdges()
812 Status ColocationGraph::AddHostOnlyDataTypesConstraints() { in AddHostOnlyDataTypesConstraints()
916 Status ColocationGraph::AddInspectionConstraints( in AddInspectionConstraints()
929 Status ColocationGraph::Initialize() { in Initialize()
1012 Status ColocationGraph::ApplyIOColocationGroups( in ApplyIOColocationGroups()
1071 Status ColocationGraph::ColocateNodeToGroup( in ColocateNodeToGroup()
1106 Status ColocationGraph::ColocateNodes(const Node& x, const Node& y) { in ColocateNodes()
[all …]
|
| H A D | inspecting_placer.cc | 75 ColocationGraph* colocation_graph) in ColocationGraphToIOColocationGroups()
107 ColocationGraph* colocation_graph_;
148 ColocationGraph colocation_graph( in ComputeIOColocationGroups()
|
| H A D | colocation_graph.h | 214 class ColocationGraph {
219 ColocationGraph(const Graph* graph, const FunctionStack& stack,
387 TF_DISALLOW_COPY_AND_ASSIGN(ColocationGraph);
|
| H A D | placer.cc | 82 const ColocationGraph& colocation_graph) { in DumpColocationGraph()
131 ColocationGraph* colocation_graph, in AssignAndLog()
185 ColocationGraph colocation_graph(graph_, stack, flib_def_, devices_, in Run()
|