diff options
author | Richard Wei <rxwei@google.com> | 2018-07-30 16:03:55 -0700 |
---|---|---|
committer | TensorFlower Gardener <gardener@tensorflow.org> | 2018-07-30 16:08:09 -0700 |
commit | f943f0c2012e3820ddfe0fa86f002a80998e1faf (patch) | |
tree | cf9af71da5275778df21cc871777859766e464fb /tensorflow/c/c_api_experimental.h | |
parent | 11b2a0beb55e14eb094b1a76952950bb314dad9f (diff) |
Add TF_CreateRunOptions. This enables Swift for TensorFlow's compiler?runtime to dump run metadata for debugging.
PiperOrigin-RevId: 206663537
Diffstat (limited to 'tensorflow/c/c_api_experimental.h')
-rw-r--r-- | tensorflow/c/c_api_experimental.h | 6 |
1 files changed, 6 insertions, 0 deletions
diff --git a/tensorflow/c/c_api_experimental.h b/tensorflow/c/c_api_experimental.h index 2d81c01e0d..6617c5a572 100644 --- a/tensorflow/c/c_api_experimental.h +++ b/tensorflow/c/c_api_experimental.h @@ -70,6 +70,12 @@ TF_CAPI_EXPORT extern TF_Buffer* TF_CreateConfig( unsigned char enable_xla_compilation, unsigned char gpu_memory_allow_growth); +// Create a serialized tensorflow.RunOptions proto, where RunOptions.trace_level +// is set to FULL_TRACE if `enable_full_trace` is non-zero, and NO_TRACE +// otherwise. +TF_CAPI_EXPORT extern TF_Buffer* TF_CreateRunOptions( + unsigned char enable_full_trace); + // Returns the graph content in a human-readable format, with length set in // `len`. The format is subject to change in the future. // The returned string is heap-allocated, and caller should call free() on it. |