Home
last modified time | relevance | path

Searched refs:UnionFind (Results 1 – 7 of 7) sorted by relevance

/external/tensorflow/tensorflow/compiler/xla/
Dunion_find.h26 class UnionFind {
28 UnionFind() : rank_(0), size_(1), parent_(nullptr) {} in UnionFind() function
35 void Merge(UnionFind* other);
43 UnionFind* FindRoot();
47 UnionFind* parent_;
52 void UnionFind<T>::Merge(UnionFind* other) { in Merge()
53 UnionFind<T>* a = FindRoot(); in Merge()
54 UnionFind<T>* b = other->FindRoot(); in Merge()
71 UnionFind<T>* UnionFind<T>::FindRoot() { in FindRoot()
/external/tensorflow/tensorflow/compiler/tf2tensorrt/segment/
Dunion_find.h151 class UnionFind {
153 UnionFind() : size_(1), parent_(nullptr) {} in UnionFind() function
154 UnionFind(const T& v, const P& p) in UnionFind() function
156 UnionFind(const T& v, P&& p) in UnionFind() function
170 Status Merge(UnionFind* other);
181 UnionFind* FindRoot();
184 UnionFind* parent_;
190 Status UnionFind<T, P>::Merge(UnionFind* other) { in Merge()
191 UnionFind<T>* a = FindRoot(); in Merge()
192 UnionFind<T>* b = other->FindRoot(); in Merge()
[all …]
Dsegment.cc673 std::vector<UnionFind<SimpleNode*>>* segments, in AddSegmentForNode()
892 std::vector<UnionFind<SimpleNode*>> node_segments; in SegmentGraph()
1001 UnionFind<SimpleNode*>* out_cluster = in SegmentGraph()
/external/mesa3d/src/amd/registers/
Dregdb.py37 class UnionFind(object): class
555 regtypes_merge = UnionFind()
556 enums_merge = UnionFind()
/external/tensorflow/tensorflow/compiler/xla/service/
Dwhile_loop_simplifier.cc352 absl::flat_hash_map<HloInstruction*, tensorflow::UnionFind<HloInstruction*>> in TryRemoveDeadWhileParams()
/external/tensorflow/tensorflow/compiler/jit/
Dmark_for_compilation_pass.cc460 std::vector<UnionFind<Cluster*>> cluster_for_node_;
/external/tensorflow/tensorflow/compiler/xla/service/gpu/
Dir_emitter_unnested.cc4854 HloInstructionMap<tensorflow::UnionFind<HloInstruction*>> disjoint_sets; in GroupDisjointReductions()