diff options
author | 2018-01-09 07:37:55 -0800 | |
---|---|---|
committer | 2018-01-09 07:41:22 -0800 | |
commit | c5e2b0e5a3039fe98b0f22154c567c2eb425fb22 (patch) | |
tree | d9184516ef5d8fe14ac4abbff97c1812bf235666 /tensorflow/python/pywrap_tfe.i | |
parent | 14fa431da0b7fd69ccf7bf4a60172a5745c1773c (diff) |
Exposes runmetadata from tfe in python.
PiperOrigin-RevId: 181317960
Diffstat (limited to 'tensorflow/python/pywrap_tfe.i')
-rw-r--r-- | tensorflow/python/pywrap_tfe.i | 3 |
1 files changed, 3 insertions, 0 deletions
diff --git a/tensorflow/python/pywrap_tfe.i b/tensorflow/python/pywrap_tfe.i index 42e4773df3..083931aa83 100644 --- a/tensorflow/python/pywrap_tfe.i +++ b/tensorflow/python/pywrap_tfe.i @@ -20,6 +20,9 @@ limitations under the License. %rename("%s") TFE_ContextListDevices; %rename("%s") TFE_ContextAddFunction; %rename("%s") TFE_ContextAddFunctionDef; +%rename("%s") TFE_ContextEnableRunMetadata; +%rename("%s") TFE_ContextDisableRunMetadata; +%rename("%s") TFE_ContextExportRunMetadata; %rename("%s") TFE_ContextClearCaches; %rename("%s") TFE_OpNameGetAttrType; %rename("%s") TFE_Py_InitEagerTensor; |