| /external/tensorflow/tensorflow/python/distribute/ |
| D | mirrored_strategy.py | 24 from tensorflow.python.distribute import cross_device_ops as cross_device_ops_lib
277 def __init__(self, devices=None, cross_device_ops=None): argument
279 self, devices=devices, cross_device_ops=cross_device_ops)
293 def __init__(self, devices=None, cross_device_ops=None): argument
295 self, devices=devices, cross_device_ops=cross_device_ops)
309 def __init__(self, container_strategy, devices=None, cross_device_ops=None): argument
329 self._cross_device_ops = cross_device_ops
744 cross_device_ops = None
746 if cross_device_ops is None:
747 cross_device_ops = self._get_cross_device_ops(value)
[all …]
|
| D | BUILD | 31 name = "cross_device_ops",
32 srcs = ["cross_device_ops.py"],
116 ":cross_device_ops",
316 ":cross_device_ops",
345 ":cross_device_ops",
409 ":cross_device_ops",
583 ":cross_device_ops",
1099 ":cross_device_ops",
|
| D | mirrored_strategy_test.py | 31 from tensorflow.python.distribute import cross_device_ops as cross_device_ops_lib
1152 cross_device_ops=cross_device_ops_lib.ReductionToOneDevice(
1297 cross_device_ops=self._make_cross_device_ops())
1305 cross_device_ops=self._make_cross_device_ops())
1329 cross_device_ops=self._make_cross_device_ops())
1337 cross_device_ops=self._make_cross_device_ops())
|
| D | parameter_server_strategy.py | 24 from tensorflow.python.distribute import cross_device_ops as cross_device_ops_lib
|
| D | collective_all_reduce_strategy.py | 29 from tensorflow.python.distribute import cross_device_ops as cross_device_ops_lib
|
| D | cross_device_ops_test.py | 33 from tensorflow.python.distribute import cross_device_ops as cross_device_ops_lib
|
| D | tpu_strategy.py | 34 from tensorflow.python.distribute import cross_device_ops as cross_device_ops_lib
|
| /external/tensorflow/tensorflow/tools/api/golden/v2/ |
| D | tensorflow.distribute.-nccl-all-reduce.pbtxt | 3 is_instance: "<class \'tensorflow.python.distribute.cross_device_ops.NcclAllReduce\'>"
4 is_instance: "<class \'tensorflow.python.distribute.cross_device_ops.AllReduceCrossDeviceOps\'>"
5 is_instance: "<class \'tensorflow.python.distribute.cross_device_ops.CrossDeviceOps\'>"
|
| D | tensorflow.distribute.-hierarchical-copy-all-reduce.pbtxt | 3 is_instance: "<class \'tensorflow.python.distribute.cross_device_ops.HierarchicalCopyAllReduce\'>"
4 is_instance: "<class \'tensorflow.python.distribute.cross_device_ops.AllReduceCrossDeviceOps\'>"
5 is_instance: "<class \'tensorflow.python.distribute.cross_device_ops.CrossDeviceOps\'>"
|
| D | tensorflow.distribute.-reduction-to-one-device.pbtxt | 3 is_instance: "<class \'tensorflow.python.distribute.cross_device_ops.ReductionToOneDevice\'>"
4 is_instance: "<class \'tensorflow.python.distribute.cross_device_ops.CrossDeviceOps\'>"
|
| D | tensorflow.distribute.-cross-device-ops.pbtxt | 3 is_instance: "<class \'tensorflow.python.distribute.cross_device_ops.CrossDeviceOps\'>"
|
| D | tensorflow.distribute.-mirrored-strategy.pbtxt | 21 …argspec: "args=[\'self\', \'devices\', \'cross_device_ops\'], varargs=None, keywords=None, default…
|
| /external/tensorflow/tensorflow/tools/api/golden/v1/ |
| D | tensorflow.distribute.-nccl-all-reduce.pbtxt | 3 is_instance: "<class \'tensorflow.python.distribute.cross_device_ops.NcclAllReduce\'>"
4 is_instance: "<class \'tensorflow.python.distribute.cross_device_ops.AllReduceCrossDeviceOps\'>"
5 is_instance: "<class \'tensorflow.python.distribute.cross_device_ops.CrossDeviceOps\'>"
|
| D | tensorflow.distribute.-hierarchical-copy-all-reduce.pbtxt | 3 is_instance: "<class \'tensorflow.python.distribute.cross_device_ops.HierarchicalCopyAllReduce\'>"
4 is_instance: "<class \'tensorflow.python.distribute.cross_device_ops.AllReduceCrossDeviceOps\'>"
5 is_instance: "<class \'tensorflow.python.distribute.cross_device_ops.CrossDeviceOps\'>"
|
| D | tensorflow.distribute.-reduction-to-one-device.pbtxt | 3 is_instance: "<class \'tensorflow.python.distribute.cross_device_ops.ReductionToOneDevice\'>"
4 is_instance: "<class \'tensorflow.python.distribute.cross_device_ops.CrossDeviceOps\'>"
|
| D | tensorflow.distribute.-cross-device-ops.pbtxt | 3 is_instance: "<class \'tensorflow.python.distribute.cross_device_ops.CrossDeviceOps\'>"
|
| D | tensorflow.distribute.-mirrored-strategy.pbtxt | 21 …argspec: "args=[\'self\', \'devices\', \'cross_device_ops\'], varargs=None, keywords=None, default…
|
| /external/tensorflow/tensorflow/python/distribute/v1/ |
| D | cross_device_ops_test.py | 33 from tensorflow.python.distribute import cross_device_ops as cross_device_ops_lib
162 def _testReductionAndBroadcast(self, cross_device_ops, devices): argument
189 cross_device_ops.reduce(
195 cross_device_ops.reduce(
201 cross_device_ops.reduce(
207 cross_device_ops.reduce(
216 cross_device_ops.batch_reduce(reduce_util.ReduceOp.MEAN,
222 cross_device_ops.batch_reduce(reduce_util.ReduceOp.SUM,
233 cross_device_ops.broadcast(constant_op.constant(1.), destinations),
282 cross_device_ops=[
[all …]
|
| D | BUILD | 27 "//tensorflow/python/distribute:cross_device_ops",
|
| /external/tensorflow/tensorflow/python/training/ |
| D | optimizer_test.py | 21 from tensorflow.python.distribute import cross_device_ops
281 cross_device_ops=cross_device_ops.HierarchicalCopyAllReduce())
|
| D | BUILD | 1221 "//tensorflow/python/distribute:cross_device_ops",
|
| /external/tensorflow/tensorflow/python/keras/benchmarks/ |
| D | distribution_util.py | 135 cross_device_ops=_mirrored_cross_device_ops(all_reduce_alg, num_packs))
|