| /external/tensorflow/tensorflow/contrib/distribute/python/ |
| D | mirrored_strategy.py | 91 cross_device_ops=None, argument
94 assert not (cross_device_ops and cross_tower_ops)
101 cross_device_ops or cross_tower_ops,
168 cross_device_ops=None, argument
176 cross_device_ops)
|
| D | cross_device_ops_test.py | 31 from tensorflow.python.distribute import cross_device_ops as cross_device_ops_lib
130 def _testReductionAndBroadcast(self, cross_device_ops, distribution): argument
152 cross_device_ops.reduce(
158 cross_device_ops.reduce(
164 cross_device_ops.reduce(
169 cross_device_ops.reduce(
178 cross_device_ops.batch_reduce(
186 cross_device_ops.batch_reduce(
197 cross_device_ops.broadcast(constant_op.constant(1.), destinations),
206 cross_device_ops=[
[all …]
|
| D | collective_all_reduce_strategy.py | 22 from tensorflow.python.distribute import cross_device_ops as cross_device_ops_lib
|
| D | estimator_training_test.py | 37 from tensorflow.python.distribute import cross_device_ops as cross_device_ops_lib
|
| D | collective_all_reduce_strategy_test.py | 33 from tensorflow.python.distribute import cross_device_ops as cross_device_ops_lib
|
| D | BUILD | 554 "//tensorflow/python/distribute:cross_device_ops",
|
| D | mirrored_strategy_multigpu_test.py | 33 from tensorflow.python.distribute import cross_device_ops as cross_device_ops_lib
|
| /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 | 16 …argspec: "args=[\'self\', \'devices\', \'cross_device_ops\'], varargs=None, keywords=None, default…
|
| /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 | 16 …argspec: "args=[\'self\', \'devices\', \'cross_device_ops\'], varargs=None, keywords=None, default…
|
| /external/tensorflow/tensorflow/python/distribute/ |
| D | BUILD | 48 name = "cross_device_ops",
49 srcs = ["cross_device_ops.py"],
115 ":cross_device_ops",
221 ":cross_device_ops",
262 "//tensorflow/python/distribute:cross_device_ops",
301 "//tensorflow/python/distribute:cross_device_ops",
|
| D | __init__.py | 23 from tensorflow.python.distribute import cross_device_ops
|
| D | mirrored_strategy.py | 26 from tensorflow.python.distribute import cross_device_ops as cross_device_ops_lib
436 def __init__(self, devices=None, cross_device_ops=None): argument
438 self, devices=devices, cross_device_ops=cross_device_ops)
445 def __init__(self, container_strategy, devices=None, cross_device_ops=None): argument
452 self._cross_device_ops = cross_device_ops
|
| D | collective_all_reduce_strategy.py | 26 from tensorflow.python.distribute import cross_device_ops as cross_device_ops_lib
|
| D | parameter_server_strategy.py | 24 from tensorflow.python.distribute import cross_device_ops as cross_device_ops_lib
|
| D | tpu_strategy.py | 24 from tensorflow.python.distribute import cross_device_ops as cross_device_ops_lib
|
| /external/tensorflow/tensorflow/contrib/distribute/ |
| D | __init__.py | 35 from tensorflow.python.distribute.cross_device_ops import *
|
| D | BUILD | 37 "//tensorflow/python/distribute:cross_device_ops",
|