Searched refs:all_max (Results 1 – 3 of 3) sorted by relevance
423 bool all_max = true; in Optimize() local426 all_max = false; in Optimize()430 if (output_time < processing_time / cpu_budget || all_max) { in Optimize()
113 def all_max(tensors): function
136 self._Test(partial(_NcclAllReduce, nccl_ops.all_max), np.maximum)