Searched refs:ds_reduce_util (Results 1 – 4 of 4) sorted by relevance
23 from tensorflow.python.distribute import reduce_util as ds_reduce_util unknown151 return distribution.extended.batch_reduce_to(ds_reduce_util.ReduceOp.SUM,
27 from tensorflow.python.distribute import reduce_util as ds_reduce_util unknown108 reduce_op = ds_reduce_util.ReduceOp.SUM112 reduce_op = ds_reduce_util.ReduceOp.MEAN346 reduce_op = ds_reduce_util.ReduceOp.SUM350 reduce_op = ds_reduce_util.ReduceOp.MEAN
22 from tensorflow.python.distribute import reduce_util as ds_reduce_util unknown107 value = strategy.extended.reduce_to(ds_reduce_util.ReduceOp.MEAN, value,
30 from tensorflow.python.distribute import reduce_util as ds_reduce_util unknown532 if distribute_lib.get_loss_reduction() == ds_reduce_util.ReduceOp.MEAN:676 ds_reduce_util.ReduceOp.SUM, grads_and_vars)