diff options
author | 2017-11-27 13:42:53 -0800 | |
---|---|---|
committer | 2017-11-27 13:46:22 -0800 | |
commit | db9533e4f5fa940f704996cd6d38f40b13d40dff (patch) | |
tree | 2d9ace33f9451e3ae745736027d1f83d1bd34fb5 | |
parent | 2110de0059a139ed5f4dd15c3e79102cceecad74 (diff) |
Enable a Session tests using function and set_device
These tests were disabled for C API because those features
were not implemented. They are now.
PiperOrigin-RevId: 177064985
-rw-r--r-- | tensorflow/python/client/session_test.py | 3 |
1 files changed, 0 insertions, 3 deletions
diff --git a/tensorflow/python/client/session_test.py b/tensorflow/python/client/session_test.py index 6b45a5f313..3e85410a97 100644 --- a/tensorflow/python/client/session_test.py +++ b/tensorflow/python/client/session_test.py @@ -1583,7 +1583,6 @@ class SessionTest(test_util.TensorFlowTestCase): sess.run(enqueue_op) self.assertEqual(sess.run(q.size()), num_epochs * 2) - @test_util.disable_c_api # set_device does not work with C API def testRegisterFetchAndFeedConversionFunctions(self): class SquaredTensor(object): def __init__(self, tensor): @@ -1733,11 +1732,9 @@ class SessionTest(test_util.TensorFlowTestCase): result = sess.run(f) self.assertEqual(result, 2.0) - @test_util.disable_c_api # functions don't work with C API def testAddFunctionToSession(self): self.runTestAddFunctionToSession() - @test_util.disable_c_api # functions don't work with C API def testAddFunctionToGrpcSession(self): server = server_lib.Server.create_local_server() self.runTestAddFunctionToSession(server.target) |