Home
last modified time | relevance | path

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

1234

/external/tensorflow/tensorflow/python/distribute/
Done_device_strategy.py18 from tensorflow.python.distribute import distribute_lib
38 class OneDeviceStrategy(distribute_lib.Strategy):
78 distribute_lib.distribution_strategy_gauge.get_cell("V2").set(
239 class OneDeviceStrategyV1(distribute_lib.StrategyV1):
247 distribute_lib.distribution_strategy_gauge.get_cell("V1").set(
253 class OneDeviceExtended(distribute_lib.StrategyExtendedV1):
297 replication_mode=distribute_lib.InputReplicationMode.PER_WORKER):
299 [distribute_lib.InputContext()],
314 distribute_lib.InputReplicationMode.PER_REPLICA):
328 distribute_lib.InputReplicationMode.PER_REPLICA):
[all …]
Dcentral_storage_strategy.py18 from tensorflow.python.distribute import distribute_lib
24 class CentralStorageStrategy(distribute_lib.Strategy):
66 distribute_lib.distribution_strategy_gauge.get_cell('V2').set(
102 distribute_lib.InputReplicationMode.PER_REPLICA):
213 class CentralStorageStrategyV1(distribute_lib.StrategyV1):
223 distribute_lib.distribution_strategy_gauge.get_cell('V1').set(
Ddistribution_strategy_context.py28 distribute_lib = LazyLoader( variable
373 _ = distribute_lib._creating_default_strategy_singleton
374 distribute_lib._creating_default_strategy_singleton = True
376 _defaults["strategy"] = distribute_lib._DefaultDistributionStrategy()
379 distribute_lib._DefaultDistributionStrategyV1())
380 distribute_lib._creating_default_strategy_singleton = False
391 _defaults["replica_context"] = distribute_lib._DefaultReplicaContext(
Dtpu_strategy.py32 from tensorflow.python.distribute import distribute_lib
230 class TPUStrategyV2(distribute_lib.Strategy):
359 distribute_lib.distribution_strategy_gauge.get_cell("V2").set("TPUStrategy")
360 distribute_lib.distribution_strategy_replica_gauge.get_cell(
362 distribute_lib.distribution_strategy_replica_gauge.get_cell(
427 options = options or distribute_lib.RunOptions()
656 class TPUStrategy(distribute_lib.Strategy):
698 distribute_lib.distribution_strategy_gauge.get_cell("V2").set("TPUStrategy")
699 distribute_lib.distribution_strategy_replica_gauge.get_cell(
701 distribute_lib.distribution_strategy_replica_gauge.get_cell(
[all …]
Ddistribute_lib_test.py22 from tensorflow.python.distribute import distribute_lib
39 class _TestReplicaContext(distribute_lib.ReplicaContext):
58 class _TestStrategy(distribute_lib.Strategy):
64 class _TestExtended(distribute_lib.StrategyExtendedV1):
83 replication_mode=distribute_lib.InputReplicationMode.PER_WORKER):
85 [distribute_lib.InputContext()],
89 return dataset_fn(distribute_lib.InputContext())
445 class _TestStrategy2(distribute_lib.Strategy):
539 dataset_fn(distribute_lib.InputContext()))
544 dataset_fn(distribute_lib.InputContext()))
[all …]
Dparameter_server_strategy.py22 from tensorflow.python.distribute import distribute_lib
49 class ParameterServerStrategyV1(distribute_lib.StrategyV1):
115 distribute_lib.distribution_strategy_gauge.get_cell("V1").set(
120 distribute_lib.InputReplicationMode.PER_REPLICA):
133 distribute_lib.InputReplicationMode.PER_REPLICA):
160 class ParameterServerStrategyExtended(distribute_lib.StrategyExtendedV1):
357 replication_mode=distribute_lib.InputReplicationMode.PER_WORKER):
367 input_context = distribute_lib.InputContext(
389 input_context = distribute_lib.InputContext(
404 value_fn(distribute_lib.ValueContext(replica_id,
[all …]
Dcollective_all_reduce_strategy.py28 from tensorflow.python.distribute import distribute_lib
58 class CollectiveAllReduceStrategy(distribute_lib.Strategy):
192 distribute_lib.distribution_strategy_gauge.get_cell("V2").set(
195 distribute_lib.distribution_strategy_replica_gauge.get_cell(
197 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(
336 distribute_lib.StrategyExtendedV1.__init__(self, container_strategy)
[all …]
Dinput_lib_test.py30 from tensorflow.python.distribute import distribute_lib
84 distribute_lib.InputContext(
278 return input_fn(distribute_lib.InputContext())
300 dataset_fn(distribute_lib.InputContext()), input_workers, distribution)
1109 ctx = ctx or distribute_lib.InputContext()
1229 ctx = ctx or distribute_lib.InputContext()
1244 dataset_fn(distribute_lib.InputContext()))
1272 ctx = ctx or distribute_lib.InputContext()
1288 dataset_fn(distribute_lib.InputContext()))
1531 distribute_lib.InputOptions(
[all …]
Dmirrored_strategy.py24 from tensorflow.python.distribute import distribute_lib
200 class MirroredStrategy(distribute_lib.Strategy):
289 distribute_lib.distribution_strategy_gauge.get_cell("V2").set(
294 class MirroredStrategyV1(distribute_lib.StrategyV1): # pylint: disable=g-missing-docstring
305 distribute_lib.distribution_strategy_gauge.get_cell("V1").set(
310 class MirroredExtended(distribute_lib.StrategyExtendedV1):
450 distribute_lib.InputReplicationMode.PER_REPLICA):
552 replication_mode=distribute_lib.InputReplicationMode.PER_WORKER):
556 input_contexts.append(distribute_lib.InputContext(
566 distribute_lib.InputReplicationMode.PER_REPLICA):
[all …]
DBUILD121 ":distribute_lib",
136 name = "distribute_lib",
138 "distribute_lib.py",
172 ":distribute_lib",
253 ":distribute_lib",
284 ":distribute_lib",
330 ":distribute_lib",
361 ":distribute_lib",
392 ":distribute_lib",
404 ":distribute_lib",
[all …]
Dinput_lib_type_spec_test.py26 from tensorflow.python.distribute import distribute_lib
292 options = distribute_lib.InputOptions(
297 distribute_lib.InputReplicationMode.PER_REPLICA))
415 ctx = distribute_lib.InputContext()
477 options = distribute_lib.InputOptions(
543 options = distribute_lib.InputOptions(
548 distribute_lib.InputReplicationMode.PER_REPLICA))
589 ctx = distribute_lib.InputContext()
650 ctx = distribute_lib.InputContext()
671 options = distribute_lib.InputOptions(experimental_fetch_to_device=False)
[all …]
Dparameter_server_strategy_v2.py26 from tensorflow.python.distribute import distribute_lib
68 class ParameterServerStrategyV2(distribute_lib.Strategy):
483 distribute_lib.distribution_strategy_gauge.get_cell("V2").set(
524 distribute_lib.distribution_strategy_replica_gauge.get_cell(
526 distribute_lib.distribution_strategy_replica_gauge.get_cell(
568 distribute_lib.distribution_strategy_replica_gauge.get_cell(
890 input_context = distribute_lib.InputContext(
/external/tensorflow/tensorflow/python/tpu/tests/
Dtpu_embedding_v2_hd_valid_input_test.py22 from tensorflow.python.distribute import distribute_lib
40 options=distribute_lib.InputOptions(
47 options=distribute_lib.InputOptions(
54 options=distribute_lib.InputOptions(
83 options=distribute_lib.InputOptions(
94 options=distribute_lib.InputOptions(
Dtpu_embedding_v2_valid_input_test.py21 from tensorflow.python.distribute import distribute_lib
50 options=distribute_lib.InputOptions(
101 options=distribute_lib.InputOptions(
106 options=distribute_lib.InputOptions(
130 options=distribute_lib.InputOptions(
187 options=distribute_lib.InputOptions(
272 options=distribute_lib.InputOptions(
Dtpu_embedding_v2_enqueue_mode_test.py20 from tensorflow.python.distribute import distribute_lib
47 options=distribute_lib.InputOptions(
86 options=distribute_lib.InputOptions(
119 options=distribute_lib.InputOptions(
147 options=distribute_lib.InputOptions(
Dtpu_embedding_v2_invalid_input_test.py19 from tensorflow.python.distribute import distribute_lib
83 options=distribute_lib.InputOptions(
107 options=distribute_lib.InputOptions(
112 options=distribute_lib.InputOptions(
151 options=distribute_lib.InputOptions(
Dtpu_embedding_v2_sequence_feature_test.py19 from tensorflow.python.distribute import distribute_lib
42 options=distribute_lib.InputOptions(
73 options=distribute_lib.InputOptions(
Dtpu_embedding_v2_hd_invalid_input_test.py18 from tensorflow.python.distribute import distribute_lib
44 options=distribute_lib.InputOptions(
/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\'>"
/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\'>"
/external/tensorflow/tensorflow/python/keras/utils/
Ddataset_creator.py19 from tensorflow.python.distribute import distribute_lib
87 distribute_lib.InputOptions)):
/external/tensorflow/tensorflow/python/training/experimental/
DBUILD21 "//tensorflow/python/distribute:distribute_lib",
39 "//tensorflow/python/distribute:distribute_lib",
62 "//tensorflow/python/distribute:distribute_lib",

1234