diff options
| author |  Akshay Modi <nareshmodi@google.com> | 2018-09-06 20:58:12 -0700 |
|---|---|---|
| committer |  TensorFlower Gardener <gardener@tensorflow.org> | 2018-09-06 21:01:19 -0700 |
| commit | 38214447164a80f0c4a2ac42817ca383bd615fc0 (patch) | |
| tree | 151806eefeb7f8de7613210573624af94d97a7a0 /tensorflow/python/pywrap_tfe.i | |
| parent | 44efcf0db7b9204a77710a7f076c904d0e13e6fa (diff) | |
Set Vspace only one time
I don't believe there is currently a use-case for a different VSpace (and it doesn't seem to be controllable through any public method).
If it is a usecase we want to support, it should be simple enough to add an overload of TFE_Py_TapeGradient.
PiperOrigin-RevId: 211917235
Diffstat (limited to 'tensorflow/python/pywrap_tfe.i')
| -rwxr-xr-x | tensorflow/python/pywrap_tfe.i | 1 |
1 files changed, 1 insertions, 0 deletions
diff --git a/tensorflow/python/pywrap_tfe.i b/tensorflow/python/pywrap_tfe.i index a31861ae40..2253edc742 100755 --- a/tensorflow/python/pywrap_tfe.i +++ b/tensorflow/python/pywrap_tfe.i @@ -65,6 +65,7 @@ limitations under the License. %rename("%s") TFE_Py_TensorShapeOnDevice; %rename("%s") TFE_ContextStartStep; %rename("%s") TFE_ContextEndStep; +%rename("%s") TFE_Py_RegisterVSpace; %{ #include "tensorflow/python/eager/pywrap_tfe.h" |