diff options
| author |  Skye Wanderman-Milne <skyewm@google.com> | 2018-04-24 15:45:50 -0700 |
|---|---|---|
| committer |  TensorFlower Gardener <gardener@tensorflow.org> | 2018-04-24 15:48:41 -0700 |
| commit | e7db82f821a1c522eed9e0c633df8b3db26ef38d (patch) | |
| tree | e18a1e5afcbe56831bbce78784e0f36161921aaa /tensorflow/python/client | |
| parent | 184c8306a4a3d41f42f077b4898933500d61ce86 (diff) | |
Make TF functions work with _USE_C_SHAPES=True.
It turns out regular functions need to manually copy handle data in
addition to eager GraphModeFunctions, so I moved the C extensions to
python_api.h from eager/c_api.h.
This also cleans up function_test.py to assume the C API is enabled.
PiperOrigin-RevId: 194158700
Diffstat (limited to 'tensorflow/python/client')
| -rw-r--r-- | tensorflow/python/client/tf_session.i | 2 |
1 files changed, 1 insertions, 1 deletions
diff --git a/tensorflow/python/client/tf_session.i b/tensorflow/python/client/tf_session.i index b82182d5d3..1db1432d65 100644 --- a/tensorflow/python/client/tf_session.i +++ b/tensorflow/python/client/tf_session.i @@ -458,7 +458,7 @@ TF_ImportGraphDefResultsMissingUnusedInputMappings_wrapper{ } // Override default py3 behavior of attempting to encode into Unicode. -%typemap(out) std::string tensorflow::ResourceHandleShapeAndType { +%typemap(out) std::string tensorflow::GetResourceHandleShapeAndType { $result = PyBytes_FromStringAndSize($1.data(), $1.size()); } |