Home
last modified time | relevance | path

Searched refs:distribute_lib (Results 1 – 25 of 69) sorted by relevance

123

/external/tensorflow/tensorflow/python/distribute/
Done_device_strategy.py22 from tensorflow.python.distribute import distribute_lib
40 class OneDeviceStrategy(distribute_lib.Strategy):
80 distribute_lib.distribution_strategy_gauge.get_cell("V2").set(
241 class OneDeviceStrategyV1(distribute_lib.StrategyV1):
249 distribute_lib.distribution_strategy_gauge.get_cell("V1").set(
255 class OneDeviceExtended(distribute_lib.StrategyExtendedV1):
299 replication_mode=distribute_lib.InputReplicationMode.PER_WORKER):
301 [distribute_lib.InputContext()],
316 distribute_lib.InputReplicationMode.PER_REPLICA):
329 distribute_lib.InputReplicationMode.PER_REPLICA):
[all …]
Dcentral_storage_strategy.py22 from tensorflow.python.distribute import distribute_lib
28 class CentralStorageStrategy(distribute_lib.Strategy):
70 distribute_lib.distribution_strategy_gauge.get_cell('V2').set(
106 distribute_lib.InputReplicationMode.PER_REPLICA):
217 class CentralStorageStrategyV1(distribute_lib.StrategyV1):
227 distribute_lib.distribution_strategy_gauge.get_cell('V1').set(
Ddistribution_strategy_context.py32 distribute_lib = LazyLoader( variable
320 _ = distribute_lib._creating_default_strategy_singleton
321 distribute_lib._creating_default_strategy_singleton = True
323 _defaults["strategy"] = distribute_lib._DefaultDistributionStrategy()
326 distribute_lib._DefaultDistributionStrategyV1())
327 distribute_lib._creating_default_strategy_singleton = False
338 _defaults["replica_context"] = distribute_lib._DefaultReplicaContext(
Dtpu_strategy.py36 from tensorflow.python.distribute import distribute_lib
225 class TPUStrategyV2(distribute_lib.Strategy):
336 distribute_lib.distribution_strategy_gauge.get_cell("V2").set("TPUStrategy")
337 distribute_lib.distribution_strategy_replica_gauge.get_cell(
339 distribute_lib.distribution_strategy_replica_gauge.get_cell(
401 options = options or distribute_lib.RunOptions()
601 class TPUStrategy(distribute_lib.Strategy):
642 distribute_lib.distribution_strategy_gauge.get_cell("V2").set("TPUStrategy")
643 distribute_lib.distribution_strategy_replica_gauge.get_cell(
645 distribute_lib.distribution_strategy_replica_gauge.get_cell(
[all …]
Ddistribute_lib_test.py26 from tensorflow.python.distribute import distribute_lib
42 class _TestReplicaContext(distribute_lib.ReplicaContext):
61 class _TestStrategy(distribute_lib.Strategy):
67 class _TestExtended(distribute_lib.StrategyExtendedV1):
86 replication_mode=distribute_lib.InputReplicationMode.PER_WORKER):
88 [distribute_lib.InputContext()],
92 return dataset_fn(distribute_lib.InputContext())
428 class _TestStrategy2(distribute_lib.Strategy):
522 dataset_fn(distribute_lib.InputContext()))
527 dataset_fn(distribute_lib.InputContext()))
[all …]
Dcollective_all_reduce_strategy.py32 from tensorflow.python.distribute import distribute_lib
57 class CollectiveAllReduceStrategy(distribute_lib.Strategy):
191 distribute_lib.distribution_strategy_gauge.get_cell("V2").set(
194 distribute_lib.distribution_strategy_replica_gauge.get_cell(
196 distribute_lib.distribution_strategy_replica_gauge.get_cell(
272 class CollectiveAllReduceStrategyV1(distribute_lib.StrategyV1):
290 distribute_lib.distribution_strategy_gauge.get_cell("V1").set(
293 distribute_lib.distribution_strategy_replica_gauge.get_cell(
295 distribute_lib.distribution_strategy_replica_gauge.get_cell(
325 distribute_lib.StrategyExtendedV1.__init__(self, container_strategy)
[all …]
Dparameter_server_strategy.py26 from tensorflow.python.distribute import distribute_lib
51 class ParameterServerStrategyV1(distribute_lib.StrategyV1):
117 distribute_lib.distribution_strategy_gauge.get_cell("V1").set(
122 distribute_lib.InputReplicationMode.PER_REPLICA):
135 distribute_lib.InputReplicationMode.PER_REPLICA):
162 class ParameterServerStrategyExtended(distribute_lib.StrategyExtendedV1):
358 replication_mode=distribute_lib.InputReplicationMode.PER_WORKER):
368 input_context = distribute_lib.InputContext(
390 input_context = distribute_lib.InputContext(
405 value_fn(distribute_lib.ValueContext(replica_id,
[all …]
Dparameter_server_strategy_v2.py27 from tensorflow.python.distribute import distribute_lib
47 class ParameterServerStrategyV2(distribute_lib.Strategy):
446 distribute_lib.distribution_strategy_gauge.get_cell("V2").set(
480 distribute_lib.distribution_strategy_replica_gauge.get_cell(
482 distribute_lib.distribution_strategy_replica_gauge.get_cell(
716 return dataset_fn(distribute_lib.InputContext())
720 with distribute_lib.ReplicaContext(
Dinput_lib_test.py34 from tensorflow.python.distribute import distribute_lib
84 distribute_lib.InputContext(
280 return input_fn(distribute_lib.InputContext())
313 dataset_or_input_fn, input_workers, [distribute_lib.InputContext()],
345 dataset_fn(distribute_lib.InputContext()), input_workers, distribution)
1045 ctx = ctx or distribute_lib.InputContext()
1166 ctx = ctx or distribute_lib.InputContext()
1181 dataset_fn(distribute_lib.InputContext()))
1214 ctx = ctx or distribute_lib.InputContext()
1230 dataset_fn(distribute_lib.InputContext()))
[all …]
Dmirrored_strategy.py27 from tensorflow.python.distribute import distribute_lib
192 class MirroredStrategy(distribute_lib.Strategy):
281 distribute_lib.distribution_strategy_gauge.get_cell("V2").set(
286 class MirroredStrategyV1(distribute_lib.StrategyV1): # pylint: disable=g-missing-docstring
297 distribute_lib.distribution_strategy_gauge.get_cell("V1").set(
302 class MirroredExtended(distribute_lib.StrategyExtendedV1):
445 distribute_lib.InputReplicationMode.PER_REPLICA):
547 replication_mode=distribute_lib.InputReplicationMode.PER_WORKER):
551 input_contexts.append(distribute_lib.InputContext(
561 distribute_lib.InputReplicationMode.PER_REPLICA):
[all …]
DBUILD117 ":distribute_lib",
130 name = "distribute_lib",
132 "distribute_lib.py",
166 ":distribute_lib",
244 ":distribute_lib",
275 ":distribute_lib",
318 ":distribute_lib",
347 ":distribute_lib",
376 ":distribute_lib",
388 ":distribute_lib",
[all …]
Dinput_lib_type_spec_test.py30 from tensorflow.python.distribute import distribute_lib
296 options = distribute_lib.InputOptions(
301 distribute_lib.InputReplicationMode.PER_REPLICA))
345 ctx = distribute_lib.InputContext()
406 ctx = distribute_lib.InputContext()
427 options = distribute_lib.InputOptions(
468 ctx = distribute_lib.InputContext()
489 options = distribute_lib.InputOptions(
Done_device_strategy_test.py23 from tensorflow.python.distribute import distribute_lib
123 input_options = distribute_lib.InputOptions(
144 input_options = distribute_lib.InputOptions(
/external/tensorflow/tensorflow/python/training/experimental/
DBUILD21 "//tensorflow/python/distribute:distribute_lib",
40 "//tensorflow/python/distribute:distribute_lib",
62 "//tensorflow/python/distribute:distribute_lib",
126 "//tensorflow/python/distribute:distribute_lib",
/external/tensorflow/tensorflow/tools/api/golden/v1/
Dtensorflow.distribute.-run-options.pbtxt3 is_instance: "<class \'tensorflow.python.distribute.distribute_lib.RunOptions\'>"
4 is_instance: "<class \'tensorflow.python.distribute.distribute_lib.RunOptions\'>"
Dtensorflow.distribute.-replica-context.pbtxt3 is_instance: "<class \'tensorflow.python.distribute.distribute_lib.ReplicaContextV1\'>"
4 is_instance: "<class \'tensorflow.python.distribute.distribute_lib.ReplicaContextBase\'>"
Dtensorflow.distribute.-strategy-extended.pbtxt3 is_instance: "<class \'tensorflow.python.distribute.distribute_lib.StrategyExtendedV1\'>"
4 is_instance: "<class \'tensorflow.python.distribute.distribute_lib.StrategyExtendedV2\'>"
Dtensorflow.distribute.-strategy.pbtxt3 is_instance: "<class \'tensorflow.python.distribute.distribute_lib.StrategyV1\'>"
4 is_instance: "<class \'tensorflow.python.distribute.distribute_lib.StrategyBase\'>"
/external/tensorflow/tensorflow/tools/api/golden/v2/
Dtensorflow.distribute.-run-options.pbtxt3 is_instance: "<class \'tensorflow.python.distribute.distribute_lib.RunOptions\'>"
4 is_instance: "<class \'tensorflow.python.distribute.distribute_lib.RunOptions\'>"
Dtensorflow.distribute.-input-options.pbtxt3 is_instance: "<class \'tensorflow.python.distribute.distribute_lib.InputOptions\'>"
4 is_instance: "<class \'tensorflow.python.distribute.distribute_lib.InputOptions\'>"
Dtensorflow.distribute.-replica-context.pbtxt3 is_instance: "<class \'tensorflow.python.distribute.distribute_lib.ReplicaContext\'>"
4 is_instance: "<class \'tensorflow.python.distribute.distribute_lib.ReplicaContextBase\'>"
Dtensorflow.distribute.experimental.-central-storage-strategy.pbtxt4 is_instance: "<class \'tensorflow.python.distribute.distribute_lib.Strategy\'>"
5 is_instance: "<class \'tensorflow.python.distribute.distribute_lib.StrategyBase\'>"
Dtensorflow.distribute.-mirrored-strategy.pbtxt4 is_instance: "<class \'tensorflow.python.distribute.distribute_lib.Strategy\'>"
5 is_instance: "<class \'tensorflow.python.distribute.distribute_lib.StrategyBase\'>"
/external/tensorflow/tensorflow/python/tpu/
Dtpu_embedding_v2_test.py30 from tensorflow.python.distribute import distribute_lib
463 options=distribute_lib.InputOptions(
552 options=distribute_lib.InputOptions(
575 options=distribute_lib.InputOptions(
579 options=distribute_lib.InputOptions(
603 options=distribute_lib.InputOptions(
607 options=distribute_lib.InputOptions(
631 options=distribute_lib.InputOptions(
635 options=distribute_lib.InputOptions(
659 options=distribute_lib.InputOptions(
[all …]
/external/tensorflow/tensorflow/python/training/
Ddistribute.py22 from tensorflow.python.distribute.distribute_lib import *

123