diff options
author | 2018-08-03 12:39:27 -0700 | |
---|---|---|
committer | 2018-08-03 12:43:59 -0700 | |
commit | 249e639d634b26d1684d0df14ab6933feeba83a3 (patch) | |
tree | f3c900664ba8a3ee8150f2bffaf49c2133d66fa2 /tensorflow/python/pywrap_tfe.i | |
parent | 401a8d4c6562b9ab591068e4e5418d8967543ef6 (diff) |
Allow setting server_def directly on TFE_Context.
Any time that the server def is updated, the context is effectively "reset" by clearing all the caches.
- Check that the FLR returned is not a nullptr instead of seg faulting.
- Consolidate caches within the context object.
PiperOrigin-RevId: 207308086
Diffstat (limited to 'tensorflow/python/pywrap_tfe.i')
-rw-r--r-- | tensorflow/python/pywrap_tfe.i | 2 |
1 files changed, 1 insertions, 1 deletions
diff --git a/tensorflow/python/pywrap_tfe.i b/tensorflow/python/pywrap_tfe.i index 5d7535cf34..1b69e0d06c 100644 --- a/tensorflow/python/pywrap_tfe.i +++ b/tensorflow/python/pywrap_tfe.i @@ -29,6 +29,7 @@ limitations under the License. %rename("%s") TFE_ContextGetDevicePlacementPolicy; %rename("%s") TFE_ContextSetThreadLocalDevicePlacementPolicy; %rename("%s") TFE_ContextSetAsyncForThread; +%rename("%s") TFE_ContextSetServerDef; %rename("%s") TFE_ContextAsyncWait; %rename("%s") TFE_ContextAsyncClearError; %rename("%s") TFE_OpNameGetAttrType; @@ -59,7 +60,6 @@ limitations under the License. %rename("%s") TFE_ContextOptionsSetConfig; %rename("%s") TFE_ContextOptionsSetDevicePlacementPolicy; %rename("%s") TFE_ContextOptionsSetAsync; -%rename("%s") TFE_ContextOptionsSetServerDef; %rename("%s") TFE_DeleteContextOptions; %rename("%s") TFE_Py_TensorShapeSlice; %rename("%s") TFE_Py_TensorShapeOnDevice; |