experiments
/
STT-tensorflow
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

Increase test coverage for CentralStorageStrategy with 1 cpu and 1 gpu case 

PiperOrigin-RevId: 310266324
Change-Id: I2761421d688ed295e3a3f1d52e2c2bd79ddb75d0
This commit is contained in:
Xinyi Wang 2020-05-06 18:05:36 -07:00 committed by TensorFlower Gardener
parent 48e57d17bd
commit 09d304b9f3
1 changed files with 2 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
tensorflow/python/distribute/strategy_combinations.py
View File

@ -265,7 +265,8 @@ def distributions_and_v1_and_v2_optimizers():
strategies_minus_tpu = [ strategies_minus_tpu = [
default_strategy, one_device_strategy, one_device_strategy_gpu, default_strategy, one_device_strategy, one_device_strategy_gpu,
mirrored_strategy_with_gpu_and_cpu, mirrored_strategy_with_two_gpus mirrored_strategy_with_gpu_and_cpu, mirrored_strategy_with_two_gpus,
central_storage_strategy_with_gpu_and_cpu
] ]
strategies_minus_default_and_tpu = [ strategies_minus_default_and_tpu = [