Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/contrib/distribute/python/
Dmirrored_strategy.py30 CoreMirroredStrategy = mirrored_strategy.MirroredStrategy
35 class MirroredStrategy(distribute_lib.DistributionStrategy): class
103 super(MirroredStrategy, self).__init__(extended)
128 return super(MirroredStrategy, self).make_dataset_iterator(dataset)
157 return super(MirroredStrategy, self).experimental_make_numpy_iterator(
Destimator_training_test.py263 mirrored_strategy.MirroredStrategy,
269 mirrored_strategy.MirroredStrategy,
296 mirrored_strategy.MirroredStrategy,
323 mirrored_strategy.MirroredStrategy,
328 mirrored_strategy.MirroredStrategy,
374 None, mirrored_strategy.MirroredStrategy,
422 mirrored_strategy.MirroredStrategy,
427 mirrored_strategy.MirroredStrategy,
Dcombinations.py375 lambda: mirrored_lib.MirroredStrategy(["/cpu:0"]))
378 lambda: mirrored_lib.MirroredStrategy(["/gpu:0"]),
382 lambda: mirrored_lib.MirroredStrategy(["/gpu:0", "/cpu:0"]),
386 lambda: mirrored_lib.MirroredStrategy(["/gpu:0", "/gpu:1"]),
Dkeras_multi_worker_test.py290 if (strategy_cls == mirrored_strategy.MirroredStrategy or
311 mirrored_strategy.MirroredStrategy,
338 mirrored_strategy.MirroredStrategy,
Dcross_device_ops_test.py396 lambda: mirrored_strategy.MirroredStrategy(num_gpus_per_worker=0),
400 lambda: mirrored_strategy.MirroredStrategy(num_gpus_per_worker=1),
404 lambda: mirrored_strategy.MirroredStrategy(num_gpus_per_worker=2),
Dkeras_correctness_test_base.py282 mirrored_strategy.MirroredStrategy,
Dmirrored_strategy_multigpu_test.py910 lambda: mirrored_strategy.MirroredStrategy(
1429 lambda: mirrored_strategy.MirroredStrategy(num_gpus_per_worker=
1539 strategy = mirrored_strategy.MirroredStrategy(
Dkeras_backward_compat_test.py988 mirrored_strategy.MirroredStrategy,
/external/tensorflow/tensorflow/contrib/distribute/
DREADME.md17 * [`MirroredStrategy`](https://www.tensorflow.org/versions/master/api_docs/python/tf/contrib/distri…
23 This is a version of `MirroredStrategy` for multi-worker training. It uses
44 Let's see how to scale to multiple GPUs on one machine using `MirroredStrategy` with [tf.keras] (ht…
59 To distribute this Keras model on multiple GPUs using `MirroredStrategy` we
60 first instantiate a `MirroredStrategy` object.
63 distribution = tf.contrib.distribute.MirroredStrategy()
95 `MirroredStrategy`. It will take care of splitting up
110 …s/python/tf/estimator/Estimator). Let's see a simple example of it's usage with `MirroredStrategy`.
146 estimator. To use `MirroredStrategy`, all we need to do is:
148 * Create an instance of the `MirroredStrategy` class.
[all …]
D__init__.py28 from tensorflow.contrib.distribute.python.mirrored_strategy import MirroredStrategy
/external/tensorflow/tensorflow/python/keras/mixed_precision/experimental/
Dloss_scale_optimizer_test.py41 return mirrored_strategy.MirroredStrategy(['cpu:0', 'gpu:0'])
43 return mirrored_strategy.MirroredStrategy(['cpu:0'])
Dkeras_test.py129 return mirrored_strategy.MirroredStrategy(['cpu:0', 'gpu:0'])
131 return mirrored_strategy.MirroredStrategy(['cpu:0'])
Dautocast_variable_test.py54 return mirrored_strategy.MirroredStrategy(['cpu:0']).scope()
/external/tensorflow/tensorflow/tools/api/golden/v1/
Dtensorflow.distribute.-mirrored-strategy.pbtxt1 path: "tensorflow.distribute.MirroredStrategy"
3 is_instance: "<class \'tensorflow.python.distribute.mirrored_strategy.MirroredStrategy\'>"
Dtensorflow.distribute.pbtxt20 name: "MirroredStrategy"
/external/tensorflow/tensorflow/tools/api/golden/v2/
Dtensorflow.distribute.-mirrored-strategy.pbtxt1 path: "tensorflow.distribute.MirroredStrategy"
3 is_instance: "<class \'tensorflow.python.distribute.mirrored_strategy.MirroredStrategy\'>"
Dtensorflow.distribute.pbtxt20 name: "MirroredStrategy"
/external/tensorflow/tensorflow/python/kernel_tests/
Dtemplate_mirrored_strategy_test.py44 strategy = mirrored_strategy.MirroredStrategy(["/cpu:0", "/gpu:0"])
/external/tensorflow/tensorflow/contrib/distribute/python/examples/
Dkeras_model_with_estimator.py58 strategy = tf.contrib.distribute.MirroredStrategy(
Dsimple_estimator_example.py65 distribution = tf.contrib.distribute.MirroredStrategy(
Dkeras_mnist.py116 strategy = mirrored_strategy.MirroredStrategy(['/gpu:0', '/cpu:0'])
Dmnist_eager_multigpu.py97 strategy = tf.distribute.MirroredStrategy(devices)
/external/tensorflow/tensorflow/python/distribute/
Dmirrored_strategy.py422 class MirroredStrategy(distribute_lib.DistributionStrategy): class
439 super(MirroredStrategy, self).__init__(extended)
/external/tensorflow/tensorflow/python/training/tracking/
Dutil_with_v1_optimizers_test.py296 strategy = mirrored_strategy.MirroredStrategy()
327 strategy = mirrored_strategy.MirroredStrategy()
/external/tensorflow/
DRELEASE.md480 * Can now pass `tf.contrib.distribute.MirroredStrategy()` to `tf.estimator.RunConfig()` to run an E…