diff options
author | Kiril Gorovoy <kgorovoy@google.com> | 2017-01-27 09:06:29 -0800 |
---|---|---|
committer | TensorFlower Gardener <gardener@tensorflow.org> | 2017-01-27 09:24:36 -0800 |
commit | 4c420657c8b854354c21d30ae78624c30151eb25 (patch) | |
tree | 5fd9eec8f54593b785be680f3b2fae882b054fae /tensorflow/c/c_api_test.cc | |
parent | 54b2180f996d69ca6558cdc07939a5887ba52d80 (diff) |
Add additional named signatures to test half_plus_two model.
Change: 145801253
Diffstat (limited to 'tensorflow/c/c_api_test.cc')
-rw-r--r-- | tensorflow/c/c_api_test.cc | 3 |
1 files changed, 1 insertions, 2 deletions
diff --git a/tensorflow/c/c_api_test.cc b/tensorflow/c/c_api_test.cc index 22026f81aa..5591409d99 100644 --- a/tensorflow/c/c_api_test.cc +++ b/tensorflow/c/c_api_test.cc @@ -996,8 +996,7 @@ TEST(CAPI, SavedModel) { // Retrieve the regression signature from meta graph def. const auto signature_def_map = metagraph_def.signature_def(); - const auto signature_def = - signature_def_map.at(tensorflow::kRegressMethodName); + const auto signature_def = signature_def_map.at("regress_x_to_y"); const string input_name = signature_def.inputs().at(tensorflow::kRegressInputs).name(); |