From 09d304b9f3749d0fe51ecd340786c198907ce77d Mon Sep 17 00:00:00 2001 From: Xinyi Wang Date: Wed, 6 May 2020 18:05:36 -0700 Subject: [PATCH] Increase test coverage for CentralStorageStrategy with 1 cpu and 1 gpu case PiperOrigin-RevId: 310266324 Change-Id: I2761421d688ed295e3a3f1d52e2c2bd79ddb75d0 --- tensorflow/python/distribute/strategy_combinations.py | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) diff --git a/tensorflow/python/distribute/strategy_combinations.py b/tensorflow/python/distribute/strategy_combinations.py index 8d721698d5c..e69c8c7f129 100644 --- a/tensorflow/python/distribute/strategy_combinations.py +++ b/tensorflow/python/distribute/strategy_combinations.py @@ -265,7 +265,8 @@ def distributions_and_v1_and_v2_optimizers(): strategies_minus_tpu = [ 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 = [