diff options
author | Yuefeng Zhou <yuefengz@google.com> | 2018-04-18 16:35:44 -0700 |
---|---|---|
committer | TensorFlower Gardener <gardener@tensorflow.org> | 2018-04-18 16:38:07 -0700 |
commit | fddfa9f8dcd1a922ade5362c0538ca39e99472a7 (patch) | |
tree | 7320668c20041c85bcce4e0c6614f7270dce44c6 /tensorflow/contrib/distribute/python/mirrored_strategy_multigpu_test.py | |
parent | e9d47fbff0d644a75c6f3dcdcb852685ef515b64 (diff) |
Change distribution.distribute_dataset to accept an input_fn instead of a dataset.
PiperOrigin-RevId: 193437651
Diffstat (limited to 'tensorflow/contrib/distribute/python/mirrored_strategy_multigpu_test.py')
-rw-r--r-- | tensorflow/contrib/distribute/python/mirrored_strategy_multigpu_test.py | 4 |
1 files changed, 2 insertions, 2 deletions
diff --git a/tensorflow/contrib/distribute/python/mirrored_strategy_multigpu_test.py b/tensorflow/contrib/distribute/python/mirrored_strategy_multigpu_test.py index 59cd6703b9..6c5c055070 100644 --- a/tensorflow/contrib/distribute/python/mirrored_strategy_multigpu_test.py +++ b/tensorflow/contrib/distribute/python/mirrored_strategy_multigpu_test.py @@ -247,9 +247,9 @@ class MirroredStrategyVariableCreationTest(test.TestCase): dist = mirrored_strategy.MirroredStrategy( ["/device:GPU:0", "/device:CPU:0"]) - features = dataset_ops.Dataset.from_tensors([[1.]]).repeat(10) features = dist.distribute_dataset( - features).make_one_shot_iterator().get_next() + lambda: dataset_ops.Dataset.from_tensors([[1.]]).repeat(10) + ).make_one_shot_iterator().get_next() with dist.scope(): result = dist.call_for_each_tower( |