Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/compiler/xla/service/
Dhlo_query.h78 bool ContainsLayoutConstrainedAllReduce(const HloModule& module);
Dhlo_query.cc124 bool ContainsLayoutConstrainedAllReduce(const HloModule& module) { in ContainsLayoutConstrainedAllReduce() function
Dall_reduce_combiner.cc276 if (hlo_query::ContainsLayoutConstrainedAllReduce(*module)) { in Run()