Home
last modified time | relevance | path

Searched refs:FindRoot (Results 1 – 4 of 4) sorted by relevance

/external/tensorflow/tensorflow/compiler/tf2tensorrt/segment/
Dunion_find.h34 int Size() { return FindRoot()->size_; } in Size()
42 T& ParentValue() { return FindRoot()->value_; } in ParentValue()
49 UnionFind* FindRoot();
58 UnionFind<T>* a = FindRoot(); in Merge()
59 UnionFind<T>* b = other->FindRoot(); in Merge()
67 UnionFind<T>* UnionFind<T>::FindRoot() { in FindRoot() function
71 parent_ = parent_->FindRoot(); in FindRoot()
/external/tensorflow/tensorflow/compiler/jit/
Dunion_find.h31 int Size() { return FindRoot()->size_; } in Size()
39 T& Get() { return FindRoot()->value_; } in Get()
43 UnionFind* FindRoot();
53 UnionFind<T>* a = FindRoot(); in Merge()
54 UnionFind<T>* b = other->FindRoot(); in Merge()
71 UnionFind<T>* UnionFind<T>::FindRoot() { in FindRoot() function
75 parent_ = parent_->FindRoot(); in FindRoot()
/external/tensorflow/tensorflow/core/common_runtime/
Dcolocation_graph.h65 static int FindRoot(std::vector<Member>* tree, int node_id);
240 int FindRoot(int node_id) { return Member::FindRoot(&members_, node_id); } in FindRoot() function
Dcolocation_graph.cc293 int Member::FindRoot(std::vector<Member>* tree, int node_id) { in FindRoot() function in tensorflow::Member
298 member.parent_ = FindRoot(tree, member.parent_); in FindRoot()
530 int src_root_id = FindRoot(src->id()); in ColocateResourceOrRefEdge()
531 int dst_root_id = FindRoot(dst->id()); in ColocateResourceOrRefEdge()
613 int x_root = FindRoot(x.id()); in ColocateNodes()
614 int y_root = FindRoot(y.id()); in ColocateNodes()
680 int root = FindRoot(node.id()); in LimitToAssignedDevice()
695 const int node_root = FindRoot(node->id()); in GetDevicesForNode()
848 int node_root = FindRoot(node->id()); in DebugString()
876 if (FindRoot(id) != node_root) { in DebugInfo()