Searched refs:batch_reduce (Results 1 – 13 of 13) sorted by relevance
| /external/tensorflow/tensorflow/python/distribute/v1/ |
| D | cross_device_ops_test.py | 216 cross_device_ops.batch_reduce(reduce_util.ReduceOp.MEAN,
222 cross_device_ops.batch_reduce(reduce_util.ReduceOp.SUM,
237 reduce_op, batch_reduce): argument
245 if batch_reduce:
246 result = cross_device_ops_instance.batch_reduce(
271 if batch_reduce:
396 batch_reduce=[True, False],
400 batch_reduce): argument
403 reduce_op, batch_reduce)
416 batch_reduce=[True, False],
[all …]
|
| /external/tensorflow/tensorflow/tools/api/golden/v2/ |
| D | tensorflow.distribute.-cross-device-ops.pbtxt | 10 name: "batch_reduce"
|
| D | tensorflow.distribute.-reduction-to-one-device.pbtxt | 11 name: "batch_reduce"
|
| D | tensorflow.distribute.-nccl-all-reduce.pbtxt | 12 name: "batch_reduce"
|
| D | tensorflow.distribute.-hierarchical-copy-all-reduce.pbtxt | 12 name: "batch_reduce"
|
| /external/tensorflow/tensorflow/tools/api/golden/v1/ |
| D | tensorflow.distribute.-cross-device-ops.pbtxt | 10 name: "batch_reduce"
|
| D | tensorflow.distribute.-reduction-to-one-device.pbtxt | 11 name: "batch_reduce"
|
| D | tensorflow.distribute.-nccl-all-reduce.pbtxt | 12 name: "batch_reduce"
|
| D | tensorflow.distribute.-hierarchical-copy-all-reduce.pbtxt | 12 name: "batch_reduce"
|
| /external/tensorflow/tensorflow/python/distribute/ |
| D | cross_device_ops_test.py | 290 reduced_values = collective.batch_reduce(options.reduce_op,
751 reduced = collective.batch_reduce(reduce_util.ReduceOp.SUM,
758 reduced = collective.batch_reduce(reduce_util.ReduceOp.SUM, [(v1, v1),
792 reduced = collective.batch_reduce(reduce_util.ReduceOp.SUM, [(v, v),
875 return collective.batch_reduce(reduce_util.ReduceOp.SUM,
954 return collective.batch_reduce(reduce_util.ReduceOp.SUM,
|
| D | cross_device_ops.py | 381 def batch_reduce(self, reduce_op, value_destination_pairs, options=None): member in CrossDeviceOps
914 return self._simple_cross_replica_ops.batch_reduce(
|
| D | parameter_server_strategy.py | 527 return self._cross_device_ops.batch_reduce(reduce_op,
|
| D | mirrored_strategy.py | 751 return cross_device_ops.batch_reduce(
|