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 | 148 return distribution.extended.batch_reduce_to(ds_reduce_util.ReduceOp.SUM,
|
| /external/tensorflow/tensorflow/tools/api/golden/v1/ |
| D | tensorflow.distribute.-strategy-extended.pbtxt | 39 name: "batch_reduce_to"
|
| /external/tensorflow/tensorflow/python/distribute/ |
| D | distribute_lib_test.py | 336 x_r, y_r = dist.extended.batch_reduce_to(op,
380 x_r, y_r = dist.extended.batch_reduce_to(reduce_util.ReduceOp.MEAN,
|
| D | strategy_common_test.py | 707 reduced = strategy.extended.batch_reduce_to(reduce_util.ReduceOp.SUM,
725 reduced = strategy.extended.batch_reduce_to(reduce_util.ReduceOp.SUM,
|
| D | distribute_lib.py | 2389 def batch_reduce_to(self, reduce_op, value_destination_pairs, options=None): member in StrategyExtendedV2
2496 return self.batch_reduce_to(reduce_op, [(v, v) for v in flat_value],
3250 return strategy.extended.batch_reduce_to(
|
| /external/tensorflow/tensorflow/python/training/ |
| D | optimizer.py | 768 reduced_grads = distribution.extended.batch_reduce_to(
|
| /external/tensorflow/tensorflow/python/distribute/v1/ |
| D | cross_device_ops_test.py | 561 return test_object.extended.batch_reduce_to(reduce_op,
|
| /external/tensorflow/ |
| D | RELEASE.md | 4568 `tf.distribute.StrategyExtended.batch_reduce_to`,
|