diff options
Diffstat (limited to 'tensorflow/python/client/session_clusterspec_prop_test.py')
-rw-r--r-- | tensorflow/python/client/session_clusterspec_prop_test.py | 3 |
1 files changed, 2 insertions, 1 deletions
diff --git a/tensorflow/python/client/session_clusterspec_prop_test.py b/tensorflow/python/client/session_clusterspec_prop_test.py index c85b22eb15..f193424133 100644 --- a/tensorflow/python/client/session_clusterspec_prop_test.py +++ b/tensorflow/python/client/session_clusterspec_prop_test.py @@ -77,7 +77,8 @@ class SessionClusterSpecPropagationTest(test_util.TensorFlowTestCase): config = config_pb2.ConfigProto(cluster_def=cluster_def) with ops.Graph().as_default() as g, ops.device('/job:worker/task:1'): - const = constant_op.constant(17) + with ops.device('/cpu:0'): + const = constant_op.constant(17) sess = session.Session(server1.target, config=config, graph=g) run_options = config_pb2.RunOptions( trace_level=config_pb2.RunOptions.FULL_TRACE) |