Searched refs:ColocationGraph (Results 1 – 3 of 3) sorted by relevance
| /external/tensorflow/tensorflow/core/common_runtime/ |
| D | colocation_graph.cc | 452 ColocationGraph::ColocationGraph(const Graph* graph, in ColocationGraph() function in tensorflow::ColocationGraph
473 Status ColocationGraph::ColocateAllNodes() { in ColocateAllNodes()
527 Status ColocationGraph::ColocateResourceOrRefEdge(Node* src, Node* dst) { in ColocateResourceOrRefEdge()
550 Status ColocationGraph::ColocateResourceAndRefEdges() { in ColocateResourceAndRefEdges()
570 Status ColocationGraph::Initialize() { in Initialize()
577 Status ColocationGraph::ColocateNodeToGroup( in ColocateNodeToGroup()
612 Status ColocationGraph::ColocateNodes(const Node& x, const Node& y) { in ColocateNodes()
621 Status ColocationGraph::ColocateNodes(const Node& x, int x_root, const Node& y, in ColocateNodes()
673 Status ColocationGraph::LimitToAssignedDevice(const Node& node) { in LimitToAssignedDevice()
692 Status ColocationGraph::GetDevicesForNode( in GetDevicesForNode()
[all …]
|
| D | colocation_graph.h | 168 class ColocationGraph {
170 ColocationGraph(const Graph* graph, const DeviceSet* device_set,
|
| D | placer.cc | 68 ColocationGraph* colocation_graph, in AssignAndLog()
113 ColocationGraph colocation_graph(graph_, devices_, default_device_, in Run()
|