diff options
| author |  Skye Wanderman-Milne <skyewm@google.com> | 2018-03-19 17:34:47 -0700 |
|---|---|---|
| committer |  TensorFlower Gardener <gardener@tensorflow.org> | 2018-03-19 17:39:00 -0700 |
| commit | 2714c07c93c2fd84480f816e0da44030a0a2bd45 (patch) | |
| tree | a37ff0e0e805c37f6c77e92d44c72cc590bbb9de /tensorflow/python/client | |
| parent | b6b4ec642a632af9abaf3ca7a2b1348ab2e94bef (diff) | |
Make _USE_C_API = True and_USE_C_SHAPES = False work with import_graph_def.
Without this change, shapes wouldn't be correctly computed for
operations created via import_graph_def.
PiperOrigin-RevId: 189670312
Diffstat (limited to 'tensorflow/python/client')
| -rw-r--r-- | tensorflow/python/client/session_test.py | 3 |
1 files changed, 1 insertions, 2 deletions
diff --git a/tensorflow/python/client/session_test.py b/tensorflow/python/client/session_test.py index 44ff440cc5..6e2640efd1 100644 --- a/tensorflow/python/client/session_test.py +++ b/tensorflow/python/client/session_test.py @@ -62,8 +62,7 @@ from tensorflow.python.util import compat ops.RegisterShape('ConstructionFails')(common_shapes.unknown_shape) -# TODO(skyewm): reenable when this works with _USE_C_SHAPES=False -# @test_util.with_c_api +@test_util.with_c_api class SessionTest(test_util.TensorFlowTestCase): def setUp(self): |