diff options
| author |  Alexandre Passos <apassos@google.com> | 2017-11-08 13:44:26 -0800 |
|---|---|---|
| committer |  Andrew Selle <aselle@andyselle.com> | 2017-11-10 16:14:37 -0800 |
| commit | 2545c4e93b7c1ee21ddb3666580ff4922630d974 (patch) | |
| tree | ed3da37ca4f30f365822785f2e8b3aa2bf26388f /tensorflow/python/pywrap_tfe.i | |
| parent | fd52578963fdc3474be30c38fa9027c1c407301b (diff) | |
Moves imperative_grad to C
Neutral-to-positive on all benchmarks. Also reduces overhead of should_record.
PiperOrigin-RevId: 175057104
Diffstat (limited to 'tensorflow/python/pywrap_tfe.i')
| -rw-r--r-- | tensorflow/python/pywrap_tfe.i | 4 |
1 files changed, 2 insertions, 2 deletions
diff --git a/tensorflow/python/pywrap_tfe.i b/tensorflow/python/pywrap_tfe.i index 637f738fed..cbacf458a0 100644 --- a/tensorflow/python/pywrap_tfe.i +++ b/tensorflow/python/pywrap_tfe.i @@ -29,7 +29,7 @@ limitations under the License. %rename("%s") TFE_Py_TapeWatch; %rename("%s") TFE_Py_TapeDeleteTrace; %rename("%s") TFE_Py_TapeRecordOperation; -%rename("%s") TFE_Py_TapeExport; +%rename("%s") TFE_Py_TapeGradient; %rename("%s") TFE_NewContextOptions; %rename("%s") TFE_ContextOptionsSetConfig; %rename("%s") TFE_ContextOptionsSetDevicePlacementPolicy; @@ -125,7 +125,7 @@ limitations under the License. SWIG_fail; } if (EagerTensor_CheckExact(elem)) { - (*$1)[i] = EagerTensorHandle(elem); + (*$1)[i] = EagerTensor_Handle(elem); } else { SWIG_exception_fail(SWIG_TypeError, "provided list of inputs contains objects other " |