Searched refs:MirroredStrategy (Results 1 – 25 of 36) sorted by relevance
12
| /external/tensorflow/tensorflow/python/distribute/ |
| D | strategy_combinations.py | 59 MirroredStrategy = _version_chooser(mirrored_lib.MirroredStrategyV1, variable
60 mirrored_lib.MirroredStrategy)
124 mirrored_lib.MirroredStrategy._collective_key_base += 100000
125 return MirroredStrategy(devices)
|
| D | remote_mirrored_strategy_eager_test.py | 44 lambda: mirrored_strategy.MirroredStrategy(get_gpus()),
|
| D | strategy_gather_test.py | 198 (mirrored_strategy.MirroredStrategy,
250 (mirrored_strategy.MirroredStrategy,
516 (mirrored_strategy.MirroredStrategy,
572 (mirrored_strategy.MirroredStrategy,
|
| D | mirrored_strategy_test.py | 645 lambda: mirrored_strategy.MirroredStrategy(
1150 lambda: mirrored_strategy.MirroredStrategy(
1251 lambda: mirrored_strategy.MirroredStrategy(
1296 strategy = mirrored_strategy.MirroredStrategy(
1303 strategy = mirrored_strategy.MirroredStrategy(
1316 strategy = mirrored_strategy.MirroredStrategy()
1328 strategy = mirrored_strategy.MirroredStrategy(
1335 strategy = mirrored_strategy.MirroredStrategy(
|
| D | mirrored_strategy.py | 192 class MirroredStrategy(distribute_lib.Strategy): class
280 super(MirroredStrategy, self).__init__(extended)
288 __doc__ = MirroredStrategy.__doc__
|
| D | README.md | 35 mirrored_strategy = tf.distribute.MirroredStrategy()
|
| D | strategy_combinations_test.py | 166 self.assertIsInstance(strategy, mirrored_strategy.MirroredStrategy)
|
| /external/tensorflow/tensorflow/tools/api/golden/v2/ |
| D | tensorflow.distribute.-mirrored-strategy.pbtxt | 1 path: "tensorflow.distribute.MirroredStrategy"
3 is_instance: "<class \'tensorflow.python.distribute.mirrored_strategy.MirroredStrategy\'>"
|
| D | tensorflow.distribute.pbtxt | 36 name: "MirroredStrategy"
|
| /external/tensorflow/tensorflow/python/training/experimental/ |
| D | loss_scale_optimizer_test.py | 49 return mirrored_strategy.MirroredStrategy(['cpu:0', 'gpu:0'])
51 return mirrored_strategy.MirroredStrategy(['cpu:0'])
274 if (isinstance(strategy, mirrored_strategy.MirroredStrategy) and
|
| D | loss_scale_test.py | 47 return mirrored_strategy.MirroredStrategy(['cpu:0', 'gpu:0'])
49 return mirrored_strategy.MirroredStrategy(['cpu:0'])
|
| D | loss_scaling_gradient_tape_test.py | 46 return mirrored_strategy.MirroredStrategy(['cpu:0', 'gpu:0'])
48 return mirrored_strategy.MirroredStrategy(['cpu:0'])
|
| /external/tensorflow/tensorflow/python/kernel_tests/ |
| D | template_mirrored_strategy_test.py | 50 strategy = mirrored_strategy.MirroredStrategy(["/cpu:0", "/gpu:0"])
|
| /external/tensorflow/tensorflow/python/keras/benchmarks/ |
| D | distribution_util.py | 133 return tf.distribute.MirroredStrategy(
|
| /external/tensorflow/tensorflow/tools/api/golden/v1/ |
| D | tensorflow.distribute.pbtxt | 20 name: "MirroredStrategy"
|
| D | tensorflow.distribute.-mirrored-strategy.pbtxt | 1 path: "tensorflow.distribute.MirroredStrategy"
|
| /external/tensorflow/tensorflow/python/keras/mixed_precision/ |
| D | loss_scale_benchmark.py | 38 return mirrored_strategy.MirroredStrategy(
|
| D | loss_scale_optimizer_test.py | 58 return mirrored_strategy.MirroredStrategy(['cpu:0', 'gpu:0'])
60 return mirrored_strategy.MirroredStrategy(['cpu:0'])
313 if (isinstance(strategy_obj, mirrored_strategy.MirroredStrategy) and
646 if (isinstance(strategy, mirrored_strategy.MirroredStrategy) and
|
| D | layer_correctness_test.py | 52 return mirrored_strategy.MirroredStrategy(['cpu:0', 'cpu:1'])
|
| D | keras_test.py | 105 return mirrored_strategy.MirroredStrategy(['cpu:0', 'gpu:0'])
107 return mirrored_strategy.MirroredStrategy(['cpu:0'])
1045 if (isinstance(strategy, mirrored_strategy.MirroredStrategy) and
1166 if (isinstance(strategy, mirrored_strategy.MirroredStrategy) and
|
| /external/tensorflow/tensorflow/core/framework/ |
| D | variable.proto | 14 // current `DistributionStrategy` (eg. With `MirroredStrategy` this would be
|
| /external/tensorflow/tensorflow/python/ops/ |
| D | stateful_random_ops_test.py | 28 from tensorflow.python.distribute.mirrored_strategy import MirroredStrategy
649 strat = MirroredStrategy(devices=["cpu:0", "cpu:1"])
674 strat = MirroredStrategy(devices=devices)
|
| /external/tensorflow/tensorflow/python/training/ |
| D | optimizer_test.py | 279 ds = mirrored_strategy.MirroredStrategy(
|
| /external/tensorflow/tensorflow/python/keras/tests/ |
| D | tracking_util_with_v1_optimizers_test.py | 291 strategy = mirrored_strategy.MirroredStrategy()
324 strategy = mirrored_strategy.MirroredStrategy()
|
| /external/tensorflow/tensorflow/python/ops/numpy_ops/g3doc/ |
| D | TensorFlow_NumPy_Keras_and_Distribution_Strategy.ipynb | 234 "strategy = tf.distribute.MirroredStrategy(gpus)"
|
12