diff options
author | Alexandre Passos <apassos@google.com> | 2018-01-05 12:38:20 -0800 |
---|---|---|
committer | TensorFlower Gardener <gardener@tensorflow.org> | 2018-01-05 12:41:54 -0800 |
commit | a17038297c82357634913603ce88994a3e3ea3bf (patch) | |
tree | f1983d7baf0479be244fc4bbce32e7f829b9fccf /tensorflow/python/pywrap_tfe.i | |
parent | 388c9b73318ad2ac4204ed67cebdcb7efcf19991 (diff) |
Clear eager kernel cache when resetting random seed.
"big hammer" required for reproducibility.
PiperOrigin-RevId: 180961787
Diffstat (limited to 'tensorflow/python/pywrap_tfe.i')
-rw-r--r-- | 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 82750e9e49..d97823c17f 100644 --- a/tensorflow/python/pywrap_tfe.i +++ b/tensorflow/python/pywrap_tfe.i @@ -20,6 +20,7 @@ limitations under the License. %rename("%s") TFE_ContextListDevices; %rename("%s") TFE_ContextAddFunction; %rename("%s") TFE_ContextAddFunctionDef; +%rename("%s") TFE_ContextClearCaches; %rename("%s") TFE_OpNameGetAttrType; %rename("%s") TFE_Py_InitEagerTensor; %rename("%s") TFE_Py_RegisterExceptionClass; |