From d0ebdb98e198a10298e041c06513272c4de1222d Mon Sep 17 00:00:00 2001 From: Ben Barsdell Date: Fri, 8 Jan 2021 16:52:04 +1100 Subject: [PATCH] Set use_gpu=True in _testTypesForSparseAdagrad --- tensorflow/python/training/training_ops_test.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/tensorflow/python/training/training_ops_test.py b/tensorflow/python/training/training_ops_test.py index 3dd1283c924..d7dae28a163 100644 --- a/tensorflow/python/training/training_ops_test.py +++ b/tensorflow/python/training/training_ops_test.py @@ -225,7 +225,7 @@ class TrainingOpsTest(TensorFlowTestCase): def _testTypesForSparseAdagrad(self, x, y, lr, grad, indices): self.setUp() - with self.session(use_gpu=False): + with self.session(use_gpu=True): var = variables.VariableV1(x) accum = variables.VariableV1(y) self.evaluate(variables.global_variables_initializer())