Searched refs:reduce_to (Results 1 – 13 of 13) sorted by relevance
| /external/tensorflow/tensorflow/tools/api/golden/v2/ |
| D | tensorflow.distribute.-strategy-extended.pbtxt | 30 name: "reduce_to"
|
| /external/tensorflow/tensorflow/tools/api/golden/v1/ |
| D | tensorflow.distribute.-strategy-extended.pbtxt | 71 name: "reduce_to"
|
| /external/tensorflow/tensorflow/python/distribute/ |
| D | distribute_lib_test.py | 334 x_r = dist.extended.reduce_to(op, x, "/CPU:0")
373 x_r = dist.extended.reduce_to(reduce_util.ReduceOp.MEAN, x, "/CPU:0")
|
| D | strategy_common_test.py | 701 reduced = strategy.extended.reduce_to(reduce_util.ReduceOp.SUM, value,
719 reduced = strategy.extended.reduce_to(reduce_util.ReduceOp.SUM, value,
|
| D | values_util.py | 327 return strategy.extended.reduce_to(reduce_op, value, destinations)
|
| D | distribute_lib.py | 2299 return self._local_results(self.reduce_to(reduce_op, value, dst))[0]
2301 def reduce_to(self, reduce_op, value, destinations, options=None): member in StrategyExtendedV2
2463 self.reduce_to(reduce_op, t, destinations=v, options=options)
|
| D | strategy_test_lib.py | 193 g = d.extended.reduce_to(
249 g = d.extended.reduce_to(
|
| D | parameter_server_strategy_v2.py | 926 self.reduce_to(reduce_op, value, destinations))[0]
|
| D | collective_all_reduce_strategy_test.py | 184 g = distribution.extended.reduce_to(
|
| D | parameter_server_strategy_test.py | 485 g = d.extended.reduce_to(
|
| /external/tensorflow/tensorflow/python/training/ |
| D | moving_averages.py | 104 value = strategy.extended.reduce_to(ds_reduce_util.ReduceOp.MEAN, value,
|
| /external/tensorflow/tensorflow/python/distribute/v1/ |
| D | cross_device_ops_test.py | 553 return test_object.extended.reduce_to(reduce_op, per_replica,
|
| /external/tensorflow/ |
| D | RELEASE.md | 4567 `tf.distribute.StrategyExtended.reduce_to`,
|