Searched refs:ContainsLayoutConstrainedAllReduce (Results 1 – 3 of 3) sorted by relevance
78 bool ContainsLayoutConstrainedAllReduce(const HloModule& module);
124 bool ContainsLayoutConstrainedAllReduce(const HloModule& module) { in ContainsLayoutConstrainedAllReduce() function
276 if (hlo_query::ContainsLayoutConstrainedAllReduce(*module)) { in Run()