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.cc | |
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.cc')
-rw-r--r-- | tensorflow/c/c_api_experimental.cc | 12 |
1 files changed, 12 insertions, 0 deletions
diff --git a/tensorflow/c/c_api_experimental.cc b/tensorflow/c/c_api_experimental.cc index 170046c802..69b3ffe2a1 100644 --- a/tensorflow/c/c_api_experimental.cc +++ b/tensorflow/c/c_api_experimental.cc @@ -84,6 +84,18 @@ TF_Buffer* TF_CreateConfig(unsigned char enable_xla_compilation, return ret; } +TF_Buffer* TF_CreateRunOptions(unsigned char enable_full_trace) { + tensorflow::RunOptions options; + if (enable_full_trace) { + options.set_trace_level(tensorflow::RunOptions::FULL_TRACE); + } else { + options.set_trace_level(tensorflow::RunOptions::NO_TRACE); + } + TF_Buffer* ret = TF_NewBuffer(); + TF_CHECK_OK(MessageToBuffer(options, ret)); + return ret; +} + const char* TF_GraphDebugString(TF_Graph* graph, size_t* len) { tensorflow::mutex_lock c(graph->mu); const auto& debug_str = graph->graph.ToGraphDefDebug().DebugString(); |