Searched refs:one_device_strategy (Results 1 – 25 of 32) sorted by relevance
12
| /external/tensorflow/tensorflow/python/distribute/ |
| D | strategy_combinations_test.py | 28 from tensorflow.python.distribute import one_device_strategy
95 strategy_combinations.one_device_strategy,
101 self.assertIsInstance(strategy, one_device_strategy.OneDeviceStrategyV1)
149 strategy_combinations.one_device_strategy,
155 self.assertIsInstance(strategy, one_device_strategy.OneDeviceStrategy)
|
| D | strategy_combinations.py | 30 from tensorflow.python.distribute import one_device_strategy as one_device_lib
219 one_device_strategy = combinations.NamedDistribution( variable
334 one_device_strategy,
342 one_device_strategy,
405 one_device_strategy, one_device_strategy_gpu,
|
| D | checkpoint_utils_test.py | 71 strategy_combinations.one_device_strategy,
105 strategy_combinations.one_device_strategy,
|
| D | zero_batch_test.py | 44 strategy_combinations.one_device_strategy,
115 strategy_combinations.one_device_strategy,
165 strategy_combinations.one_device_strategy,
|
| D | input_lib_test.py | 291 strategy_combinations.one_device_strategy,
362 strategy_combinations.one_device_strategy,
550 strategy_combinations.one_device_strategy,
818 strategy_combinations.one_device_strategy,
845 strategy_combinations.one_device_strategy,
879 strategy_combinations.one_device_strategy,
1146 strategy_combinations.one_device_strategy,
1196 strategy_combinations.one_device_strategy,
1672 strategy_combinations.one_device_strategy,
|
| D | warm_starting_util_test.py | 49 strategy_combinations.one_device_strategy,
|
| D | one_device_strategy_test.py | 35 strategy_combinations.one_device_strategy,
|
| D | input_lib_type_spec_test.py | 165 strategy_combinations.one_device_strategy,
205 strategy_combinations.one_device_strategy,
|
| D | BUILD | 121 ":one_device_strategy",
383 name = "one_device_strategy",
384 srcs = ["one_device_strategy.py"],
891 ":one_device_strategy",
915 ":one_device_strategy",
|
| D | moving_averages_test.py | 37 strategy_combinations.one_device_strategy,
|
| D | metrics_v1_test.py | 79 strategy_combinations.one_device_strategy,
|
| /external/tensorflow/tensorflow/python/keras/distribute/ |
| D | strategy_combinations.py | 33 strategy_combinations.one_device_strategy,
42 strategy_combinations.one_device_strategy,
|
| D | optimizer_combinations.py | 84 strategy_combinations_base.one_device_strategy,
95 strategy_combinations_base.one_device_strategy,
106 strategy_combinations_base.one_device_strategy,
|
| D | custom_training_loop_optimizer_test.py | 76 distribution=strategy_combinations.one_device_strategy,
|
| D | keras_metrics_test.py | 78 strategy_combinations.one_device_strategy,
|
| D | keras_premade_models_test.py | 40 strategy_combinations.one_device_strategy,
|
| D | keras_utils_test.py | 289 strategy_combinations.one_device_strategy,
323 strategy_combinations.one_device_strategy,
|
| D | saved_model_test_base.py | 55 strategy_combinations.one_device_strategy,
|
| /external/tensorflow/tensorflow/python/keras/mixed_precision/ |
| D | BUILD | 186 "//tensorflow/python/distribute:one_device_strategy",
224 "//tensorflow/python/distribute:one_device_strategy",
300 "//tensorflow/python/distribute:one_device_strategy",
|
| D | loss_scale_optimizer.py | 23 from tensorflow.python.distribute import one_device_strategy
1185 one_device_strategy.OneDeviceStrategy,
1186 one_device_strategy.OneDeviceStrategyV1,
|
| /external/tensorflow/tensorflow/compiler/mlir/tfr/examples/mnist/ |
| D | mnist_train_test.py | 25 strategy_combinations.one_device_strategy,
|
| /external/tensorflow/tensorflow/tools/api/golden/v2/ |
| D | tensorflow.__internal__.distribute.combinations.pbtxt | 52 name: "one_device_strategy"
|
| D | tensorflow.distribute.-one-device-strategy.pbtxt | 3 is_instance: "<class \'tensorflow.python.distribute.one_device_strategy.OneDeviceStrategy\'>"
|
| /external/tensorflow/tensorflow/python/training/experimental/ |
| D | BUILD | 81 "//tensorflow/python/distribute:one_device_strategy",
|
| /external/tensorflow/tensorflow/tools/api/golden/v1/ |
| D | tensorflow.distribute.-one-device-strategy.pbtxt | 3 is_instance: "<class \'tensorflow.python.distribute.one_device_strategy.OneDeviceStrategyV1\'>"
|
12