diff options
| author |  Mingsheng Hong <hongm@google.com> | 2018-07-30 18:32:40 -0700 |
|---|---|---|
| committer |  TensorFlower Gardener <gardener@tensorflow.org> | 2018-07-30 18:36:51 -0700 |
| commit | 92d16b85f9fc38348207047704c66aaeefb75bdf (patch) | |
| tree | 0a77855f77db338e0c91995b2ff5595e7d84d15e /tensorflow/c/c_api_function.cc | |
| parent | 08db4a7b13c18a149ded2e5c023e4267364e47d6 (diff) | |
Added a C API to get a TF function name.
PiperOrigin-RevId: 206684358
Diffstat (limited to 'tensorflow/c/c_api_function.cc')
| -rw-r--r-- | tensorflow/c/c_api_function.cc | 4 |
1 files changed, 4 insertions, 0 deletions
diff --git a/tensorflow/c/c_api_function.cc b/tensorflow/c/c_api_function.cc index 384e6c8cb9..a2c5a42c11 100644 --- a/tensorflow/c/c_api_function.cc +++ b/tensorflow/c/c_api_function.cc @@ -536,6 +536,10 @@ TF_Function* TF_GraphToFunction(const TF_Graph* fn_body, const char* fn_name, return tf_function; } +const char* TF_FunctionName(TF_Function* func) { + return func->fdef.signature().name().c_str(); +} + void TF_GraphCopyFunction(TF_Graph* g, const TF_Function* func, const TF_Function* grad, TF_Status* status) { if (func == nullptr) { |