diff options
author | 2018-07-30 18:32:40 -0700 | |
---|---|---|
committer | 2018-07-30 18:36:51 -0700 | |
commit | 92d16b85f9fc38348207047704c66aaeefb75bdf (patch) | |
tree | 0a77855f77db338e0c91995b2ff5595e7d84d15e /tensorflow/c/c_api_function_test.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_test.cc')
-rw-r--r-- | tensorflow/c/c_api_function_test.cc | 1 |
1 files changed, 1 insertions, 0 deletions
diff --git a/tensorflow/c/c_api_function_test.cc b/tensorflow/c/c_api_function_test.cc index f7ca219c89..bb9433ce25 100644 --- a/tensorflow/c/c_api_function_test.cc +++ b/tensorflow/c/c_api_function_test.cc @@ -193,6 +193,7 @@ class CApiFunctionTest : public ::testing::Test { ASSERT_EQ(TF_OK, TF_GetCode(s_)) << TF_Message(s_); ASSERT_NE(func_, nullptr); + ASSERT_EQ(std::string(func_name_), std::string(TF_FunctionName(func_))); TF_GraphCopyFunction(host_graph_, func_, nullptr, s_); ASSERT_EQ(TF_OK, TF_GetCode(s_)) << TF_Message(s_); } |