Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/compiler/xla/
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/compiler/tf2tensorrt/segment/
Dunion_find.h161 int Size() { return FindRoot()->size_; } in Size()
165 const P& Property() { return FindRoot()->property_; } in Property()
173 const T& ParentValue() { return FindRoot()->value_; } in ParentValue()
181 UnionFind* FindRoot();
191 UnionFind<T>* a = FindRoot(); in Merge()
192 UnionFind<T>* b = other->FindRoot(); in Merge()
204 UnionFind<T, P>* UnionFind<T, P>::FindRoot() { in FindRoot() function
208 parent_ = parent_->FindRoot(); in FindRoot()
/external/tensorflow/tensorflow/compiler/mlir/tensorflow/transforms/
Dcluster_ops_by_policy.cc222 unsigned FindRoot(unsigned id);
244 unsigned ClusteringState::FindRoot(unsigned id) { in FindRoot() function in mlir::TFDevice::ClusteringState
246 return members[id].root = FindRoot(members[id].root); in FindRoot()
253 unsigned root = FindRoot(member.root); in VerifyDominanceProperty()
271 if (it != member_ids.end() && (FindRoot(it->getSecond()) == dst_root || in VerifyDominanceProperty()
272 FindRoot(it->getSecond()) == src_root)) in VerifyDominanceProperty()
292 return it != member_ids.end() && FindRoot(it->getSecond()) == src_root; in VerifyValueConstraints()
299 if (FindRoot(member.root) == src_root) worklist.emplace_back(op); in VerifyValueConstraints()
321 unsigned a_root = FindRoot(a); in Union()
322 unsigned b_root = FindRoot(b); in Union()
[all …]
/external/tensorflow/tensorflow/core/common_runtime/
Dcolocation_graph.h78 static int FindRoot(const std::vector<Member>& tree, int node_id);
371 int FindRoot(int node_id) const { in FindRoot() function
372 return Member::FindRoot(members_, node_id); in FindRoot()
Dcolocation_graph.cc383 int Member::FindRoot(const std::vector<Member>& tree, int node_id) { in FindRoot() function in tensorflow::Member
388 return FindRoot(tree, member.parent_); in FindRoot()
1385 int node_root = FindRoot(node->id()); in DebugString()
1413 if (FindRoot(id) != node_root) { in DebugInfo()
/external/tensorflow/tensorflow/dtensor/mlir/
Dlayout_propagation_v2.cc1243 void FindRoot( in FindRoot() function
1299 FindRoot(is_updated, value, producers, &roots); in FindRootsAndEmitError()