Searched refs:distribute_lib (Results 1 – 25 of 44) sorted by relevance
12
| /external/tensorflow/tensorflow/python/distribute/ |
| D | one_device_strategy.py | 22 from tensorflow.python.distribute import distribute_lib
39 class OneDeviceStrategy(distribute_lib.DistributionStrategy):
49 class OneDeviceExtended(distribute_lib.DistributionStrategyExtended):
85 replication_mode=distribute_lib.InputReplicationMode.PER_WORKER):
87 input_fn, self._input_workers, [distribute_lib.InputContext()])
157 with ops.device(self._device), distribute_lib.UpdateContext(self._device):
209 class _OneDeviceReplicaContext(distribute_lib.ReplicaContext):
214 distribute_lib.ReplicaContext.__init__(
|
| D | distribute_lib_test.py | 21 from tensorflow.python.distribute import distribute_lib
29 class _TestReplicaContext(distribute_lib.ReplicaContext):
43 class _TestStrategy(distribute_lib.DistributionStrategy):
49 class _TestExtended(distribute_lib.DistributionStrategyExtended):
152 input_context = distribute_lib.InputContext(
159 input_context = distribute_lib.InputContext(
|
| D | BUILD | 116 ":distribute_lib",
124 name = "distribute_lib",
126 "distribute_lib.py",
156 ":distribute_lib",
223 ":distribute_lib",
276 ":distribute_lib",
358 ":distribute_lib",
476 ":distribute_lib",
487 ":distribute_lib",
|
| D | distribution_strategy_context.py | 28 distribute_lib = LazyLoader( variable
223 …_defaults["strategy"] = distribute_lib._DefaultDistributionStrategy() # pylint: disable=protected…
229 _defaults["replica_context"] = distribute_lib.ReplicaContext(
|
| D | tpu_strategy.py | 26 from tensorflow.python.distribute import distribute_lib
125 class TPUStrategy(distribute_lib.DistributionStrategy):
192 class TPUExtended(distribute_lib.DistributionStrategyExtended):
265 replication_mode=distribute_lib.InputReplicationMode.PER_WORKER):
269 input_contexts.append(distribute_lib.InputContext(
492 with ops.device(d), distribute_lib.UpdateContext(d), ops.name_scope(name):
586 with ops.device(self._host_device), distribute_lib.UpdateContext(
624 class _TPUReplicaContext(distribute_lib.ReplicaContext):
632 distribute_lib.ReplicaContext.__init__(
637 distribute_lib.require_replica_context(self)
|
| D | parameter_server_strategy.py | 26 from tensorflow.python.distribute import distribute_lib
51 class ParameterServerStrategy(distribute_lib.DistributionStrategy):
96 distribute_lib.DistributionStrategyExtended):
262 replication_mode=distribute_lib.InputReplicationMode.PER_WORKER):
272 input_context = distribute_lib.InputContext(
423 with ops.colocate_with(var), distribute_lib.UpdateContext(var.device):
434 colocate_with.device), distribute_lib.UpdateContext(colocate_with):
|
| D | collective_all_reduce_strategy.py | 29 from tensorflow.python.distribute import distribute_lib
49 class CollectiveAllReduceStrategy(distribute_lib.DistributionStrategy):
90 distribute_lib.DistributionStrategyExtended.__init__(
338 replication_mode=distribute_lib.InputReplicationMode.PER_WORKER):
345 input_context = distribute_lib.InputContext(
|
| D | mirrored_strategy.py | 28 from tensorflow.python.distribute import distribute_lib
422 class MirroredStrategy(distribute_lib.DistributionStrategy):
442 class MirroredExtended(distribute_lib.DistributionStrategyExtended):
576 replication_mode=distribute_lib.InputReplicationMode.PER_WORKER):
580 input_contexts.append(distribute_lib.InputContext(
719 with ops.device(d), distribute_lib.UpdateContext(d), ops.name_scope(name):
732 with ops.device(d), distribute_lib.UpdateContext(d), ops.name_scope(name):
908 class MirroredReplicaContext(distribute_lib.ReplicaContext):
961 distribute_lib.require_replica_context(self)
|
| D | __init__.py | 24 from tensorflow.python.distribute import distribute_lib
|
| D | values.py | 27 from tensorflow.python.distribute import distribute_lib
252 device = distribute_lib.get_update_device()
503 device = distribute_lib.get_update_device()
667 update_device = distribute_lib.get_update_device()
805 device = distribute_lib.get_update_device()
915 device = distribute_lib.get_update_device()
941 update_device = distribute_lib.get_update_device()
1481 update_device = distribute_lib.get_update_device()
|
| /external/tensorflow/tensorflow/contrib/distribute/python/ |
| D | collective_all_reduce_strategy.py | 23 from tensorflow.python.distribute import distribute_lib
29 class CollectiveAllReduceStrategy(distribute_lib.DistributionStrategy):
|
| D | parameter_server_strategy.py | 21 from tensorflow.python.distribute import distribute_lib
34 class ParameterServerStrategy(distribute_lib.DistributionStrategy):
|
| D | mirrored_strategy.py | 21 from tensorflow.python.distribute import distribute_lib
35 class MirroredStrategy(distribute_lib.DistributionStrategy):
|
| D | input_lib_test.py | 26 from tensorflow.python.distribute import distribute_lib
47 distribute_lib.InputContext(
56 dataset_fn(distribute_lib.InputContext()), input_workers,
|
| D | BUILD | 76 "//tensorflow/python/distribute:distribute_lib",
87 "//tensorflow/python/distribute:distribute_lib",
151 "//tensorflow/python/distribute:distribute_lib",
175 "//tensorflow/python/distribute:distribute_lib",
199 "//tensorflow/python/distribute:distribute_lib",
231 "//tensorflow/python/distribute:distribute_lib",
|
| /external/tensorflow/tensorflow/python/training/ |
| D | distribute.py | 22 from tensorflow.python.distribute.distribute_lib import *
|
| /external/tensorflow/tensorflow/tools/api/golden/v1/ |
| D | tensorflow.distribute.-input-context.pbtxt | 3 is_instance: "<class \'tensorflow.python.distribute.distribute_lib.InputContext\'>"
|
| D | tensorflow.distribute.-replica-context.pbtxt | 3 is_instance: "<class \'tensorflow.python.distribute.distribute_lib.ReplicaContext\'>"
|
| D | tensorflow.distribute.-strategy.pbtxt | 3 is_instance: "<class \'tensorflow.python.distribute.distribute_lib.DistributionStrategy\'>"
|
| D | tensorflow.distribute.experimental.-t-p-u-strategy.pbtxt | 4 is_instance: "<class \'tensorflow.python.distribute.distribute_lib.DistributionStrategy\'>"
|
| /external/tensorflow/tensorflow/tools/api/golden/v2/ |
| D | tensorflow.distribute.-input-context.pbtxt | 3 is_instance: "<class \'tensorflow.python.distribute.distribute_lib.InputContext\'>"
|
| D | tensorflow.distribute.-replica-context.pbtxt | 3 is_instance: "<class \'tensorflow.python.distribute.distribute_lib.ReplicaContext\'>"
|
| D | tensorflow.distribute.-strategy.pbtxt | 3 is_instance: "<class \'tensorflow.python.distribute.distribute_lib.DistributionStrategy\'>"
|
| /external/tensorflow/tensorflow/python/ops/losses/ |
| D | BUILD | 32 "//tensorflow/python/distribute:distribute_lib",
|
| /external/tensorflow/tensorflow/contrib/distribute/ |
| D | __init__.py | 38 from tensorflow.python.distribute.distribute_lib import *
|
12