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/pywrap_tfe.i | |
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/pywrap_tfe.i')
-rw-r--r-- | tensorflow/python/pywrap_tfe.i | 2 |
1 files changed, 0 insertions, 2 deletions
diff --git a/tensorflow/python/pywrap_tfe.i b/tensorflow/python/pywrap_tfe.i index 0982a67dee..5ee55301df 100644 --- a/tensorflow/python/pywrap_tfe.i +++ b/tensorflow/python/pywrap_tfe.i @@ -59,8 +59,6 @@ limitations under the License. %rename("%s") TFE_ContextOptionsSetAsync; %rename("%s") TFE_DeleteContextOptions; %rename("%s") TFE_Py_TensorShapeSlice; -%rename("%s") TFE_GetResourceHandleShapeAndType; -%rename("%s") TFE_SetResourceHandleShapeAndType; %{ #include "tensorflow/python/eager/pywrap_tfe.h" |