diff options
author | 2018-08-08 10:48:23 -0700 | |
---|---|---|
committer | 2018-08-08 10:51:51 -0700 | |
commit | 4a4ae62c75f1de3455c3adea96802d22c7e986e3 (patch) | |
tree | b499f2cb33fc2aaac2963afa9c12be2bfb8cd37a /tensorflow/python/pywrap_tfe.i | |
parent | 826e190e053abcf722cd33084ef2d31b6ed1b2aa (diff) |
Allows differentiating tfe.defun functions with loops in eager mode.
Adopts a minimal sensible policy for step containers: starting a graident tape
creates a step container; inner tapes do nothing; popping out of the outermost
tape will reset that step container. This should allow us to have reasonable
behavior in the presence of step-container-scoped things for a while. Ideally
we'll move away from them in favor of lists but the infrastructure isn't ready
yet.
PiperOrigin-RevId: 207911091
Diffstat (limited to 'tensorflow/python/pywrap_tfe.i')
-rw-r--r-- | tensorflow/python/pywrap_tfe.i | 2 |
1 files changed, 2 insertions, 0 deletions
diff --git a/tensorflow/python/pywrap_tfe.i b/tensorflow/python/pywrap_tfe.i index 1b69e0d06c..157f2341e0 100644 --- a/tensorflow/python/pywrap_tfe.i +++ b/tensorflow/python/pywrap_tfe.i @@ -63,6 +63,8 @@ limitations under the License. %rename("%s") TFE_DeleteContextOptions; %rename("%s") TFE_Py_TensorShapeSlice; %rename("%s") TFE_Py_TensorShapeOnDevice; +%rename("%s") TFE_ContextStartStep; +%rename("%s") TFE_ContextEndStep; %{ #include "tensorflow/python/eager/pywrap_tfe.h" |