Searched refs:batch_reduce_to (Results 1 – 9 of 9) sorted by relevance
| /external/tensorflow/tensorflow/tools/api/golden/v2/ |
| D | tensorflow.distribute.-strategy-extended.pbtxt | 22 name: "batch_reduce_to"
|
| /external/tensorflow/tensorflow/python/keras/optimizer_v2/ |
| D | utils.py | 151 return distribution.extended.batch_reduce_to(ds_reduce_util.ReduceOp.SUM,
|
| /external/tensorflow/tensorflow/python/distribute/ |
| D | strategy_common_test.py | 282 reduced = strategy.extended.batch_reduce_to(reduce_util.ReduceOp.SUM,
300 reduced = strategy.extended.batch_reduce_to(reduce_util.ReduceOp.SUM,
|
| D | distribute_lib_test.py | 339 x_r, y_r = dist.extended.batch_reduce_to(op,
363 x_r, y_r = dist.extended.batch_reduce_to(reduce_util.ReduceOp.MEAN,
|
| D | distribute_lib.py | 2312 def batch_reduce_to(self, reduce_op, value_destination_pairs, options=None): member in StrategyExtendedV2
2420 return self.batch_reduce_to(reduce_op, [(v, v) for v in flat_value],
3122 return strategy.extended.batch_reduce_to(
|
| /external/tensorflow/tensorflow/tools/api/golden/v1/ |
| D | tensorflow.distribute.-strategy-extended.pbtxt | 39 name: "batch_reduce_to"
|
| /external/tensorflow/tensorflow/python/training/ |
| D | optimizer.py | 675 reduced_grads = distribution.extended.batch_reduce_to(
|
| /external/tensorflow/tensorflow/python/distribute/v1/ |
| D | cross_device_ops_test.py | 562 return test_object.extended.batch_reduce_to(reduce_op,
|
| /external/tensorflow/ |
| D | RELEASE.md | 448 `tf.distribute.StrategyExtended.batch_reduce_to`, `tf.distribute.ReplicaContext.all_reduce`
|