diff options
author | Skye Wanderman-Milne <skyewm@google.com> | 2018-03-27 15:07:05 -0700 |
---|---|---|
committer | TensorFlower Gardener <gardener@tensorflow.org> | 2018-03-27 15:09:19 -0700 |
commit | 736e055a756cf0f99d59b67284aade01baec9799 (patch) | |
tree | 43034482ec323beb9d0289a4b5261a1510c1f3d7 /tensorflow/python/client | |
parent | 6bb4f7abb03a7904fecc5b61e3ed37e9b663d6b0 (diff) |
Make _USE_C_API = True and _USE_C_SHAPES = False work with handle data, take 2.
This change makes _set_shapes_for_outputs_c_api fetch and set
Tensor._handle_data. This is necessary for running the
Python shape inference code on resource tensors.
PiperOrigin-RevId: 190681459
Diffstat (limited to 'tensorflow/python/client')
-rw-r--r-- | tensorflow/python/client/tf_session.i | 1 |
1 files changed, 1 insertions, 0 deletions
diff --git a/tensorflow/python/client/tf_session.i b/tensorflow/python/client/tf_session.i index e88fc0c01a..70a3d032f4 100644 --- a/tensorflow/python/client/tf_session.i +++ b/tensorflow/python/client/tf_session.i @@ -723,6 +723,7 @@ def TF_Reset(target, containers=None, config=None): %unignore TF_TryEvaluateConstant_wrapper; %noexception TF_TryEvaluateConstant_wrapper; %unignore ExtendSession; +%unignore ResourceHandleShapeAndType; %include "tensorflow/python/client/tf_session_helper.h" |