diff options
| author |  Alexandre Passos <apassos@google.com> | 2018-01-24 09:10:53 -0800 |
|---|---|---|
| committer |  TensorFlower Gardener <gardener@tensorflow.org> | 2018-01-24 09:15:15 -0800 |
| commit | 37bfad3c33c005077630b021ca927608dd70bb3e (patch) | |
| tree | 832cc092ba6d8ebea0f16077541e854d9dc6a5c1 /tensorflow/python/pywrap_tfe.i | |
| parent | a346b737046574a86268bf920e4c88f19a455830 (diff) | |
Allow setting per-thread device copying policies in TFE.
PiperOrigin-RevId: 183093407
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 de0e5856ff..ed7e4b17ad 100644 --- a/tensorflow/python/pywrap_tfe.i +++ b/tensorflow/python/pywrap_tfe.i @@ -24,6 +24,8 @@ limitations under the License. %rename("%s") TFE_ContextDisableRunMetadata; %rename("%s") TFE_ContextExportRunMetadata; %rename("%s") TFE_ContextClearCaches; +%rename("%s") TFE_ContextGetDevicePlacementPolicy; +%rename("%s") TFE_ContextSetThreadLocalDevicePlacementPolicy; %rename("%s") TFE_OpNameGetAttrType; %rename("%s") TFE_Py_InitEagerTensor; %rename("%s") TFE_Py_RegisterExceptionClass; |