Searched refs:one_device_strategy_gpu (Results 1 – 10 of 10) sorted by relevance
| /external/tensorflow/tensorflow/python/keras/distribute/ |
| D | strategy_combinations.py | 34 strategy_combinations.one_device_strategy_gpu,
43 strategy_combinations.one_device_strategy_gpu,
|
| D | keras_premade_models_test.py | 41 strategy_combinations.one_device_strategy_gpu,
|
| D | saved_model_test_base.py | 56 strategy_combinations.one_device_strategy_gpu,
|
| D | distribute_strategy_test.py | 267 strategy_combinations.one_device_strategy_gpu,
2214 strategy_combinations.one_device_strategy_gpu,
|
| /external/tensorflow/tensorflow/python/distribute/ |
| D | strategy_combinations.py | 221 one_device_strategy_gpu = combinations.NamedDistribution( variable
335 one_device_strategy_gpu,
343 one_device_strategy_gpu,
405 one_device_strategy, one_device_strategy_gpu,
|
| D | strategy_combinations_test.py | 96 strategy_combinations.one_device_strategy_gpu,
150 strategy_combinations.one_device_strategy_gpu,
|
| D | one_device_strategy_test.py | 36 strategy_combinations.one_device_strategy_gpu
|
| D | strategy_gather_test.py | 48 strategy_combinations.one_device_strategy_gpu,
|
| /external/tensorflow/tensorflow/compiler/mlir/tfr/examples/mnist/ |
| D | mnist_train_test.py | 26 strategy_combinations.one_device_strategy_gpu,
|
| /external/tensorflow/tensorflow/tools/api/golden/v2/ |
| D | tensorflow.__internal__.distribute.combinations.pbtxt | 56 name: "one_device_strategy_gpu"
|