Searched refs:all_max (Results 1 – 3 of 3) sorted by relevance
113 def all_max(tensors): function
137 self._Test(partial(_NcclAllReduce, nccl_ops.all_max), np.maximum)
1712 bool all_max = true; in ShouldStop() local1716 all_max = false; in ShouldStop()1723 return all_max || TotalMaximumBufferedBytes(snapshot) > ram_budget; in ShouldStop()1861 bool all_max = true; in OptimizeHillClimb() local1864 all_max = false; in OptimizeHillClimb()1869 all_max || in OptimizeHillClimb()