diff options
author | Alexandre Passos <apassos@google.com> | 2017-11-30 10:09:52 -0800 |
---|---|---|
committer | TensorFlower Gardener <gardener@tensorflow.org> | 2017-11-30 10:13:24 -0800 |
commit | 12976748822cdb3885f37dbda42ce8674afa6f91 (patch) | |
tree | a073eb0515e40fb7fdab690471cbef61f8a8d326 /tensorflow/python/pywrap_tfe.i | |
parent | f283173062f3ff9b6f69e8fc8a77421dcfdaa8f2 (diff) |
Uses C API for eager functions.
Rolls back the rollback with some swiggery to get python3 to work.
PiperOrigin-RevId: 177470328
Diffstat (limited to 'tensorflow/python/pywrap_tfe.i')
-rw-r--r-- | tensorflow/python/pywrap_tfe.i | 3 |
1 files changed, 2 insertions, 1 deletions
diff --git a/tensorflow/python/pywrap_tfe.i b/tensorflow/python/pywrap_tfe.i index 82b154164e..82750e9e49 100644 --- a/tensorflow/python/pywrap_tfe.i +++ b/tensorflow/python/pywrap_tfe.i @@ -18,6 +18,7 @@ limitations under the License. %rename("%s") TFE_NewContext; %rename("%s") TFE_DeleteContext; %rename("%s") TFE_ContextListDevices; +%rename("%s") TFE_ContextAddFunction; %rename("%s") TFE_ContextAddFunctionDef; %rename("%s") TFE_OpNameGetAttrType; %rename("%s") TFE_Py_InitEagerTensor; @@ -149,7 +150,7 @@ limitations under the License. } $1 = &temp; $1->resize(PyInt_AsLong($input), nullptr); -} +} // Create new Status object. %typemap(in, numinputs=0) TF_Status *out_status { |