Home
last modified time | relevance | path

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

/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/service/
H A Dhlo_query.h85 inline bool ContainsLayoutConstrainedAllReduce(const HloModule& module) { in ContainsLayoutConstrainedAllReduce() function
H A Dall_reduce_reassociate.cc48 if (hlo_query::ContainsLayoutConstrainedAllReduce(*module)) { in Run()
H A Dall_reduce_contiguous.cc91 if (hlo_query::ContainsLayoutConstrainedAllReduce(*module)) { in Run()
H A Dall_reduce_combiner.cc125 if (hlo_query::ContainsLayoutConstrainedAllReduce(*module)) { in Run()
H A Dall_reduce_folder.cc142 if (hlo_query::ContainsLayoutConstrainedAllReduce(*module)) { in Run()
/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/service/gpu/
H A Dall_reduce_blueconnect.cc273 if (hlo_query::ContainsLayoutConstrainedAllReduce(*module)) { in Run()