diff options
author | 2017-12-12 07:43:24 -0800 | |
---|---|---|
committer | 2017-12-12 07:47:29 -0800 | |
commit | 94686be226fc999437618959a791d7bda18e1806 (patch) | |
tree | a7483bc9cdbccf17bbb574eb6022748b8885f094 /tensorflow/c/c_api_function_test.cc | |
parent | 5f470cf9b0b8adf9cf74da33e30024c62647165c (diff) |
Automated g4 rollback of changelist 178675527
PiperOrigin-RevId: 178759398
Diffstat (limited to 'tensorflow/c/c_api_function_test.cc')
-rw-r--r-- | tensorflow/c/c_api_function_test.cc | 45 |
1 files changed, 0 insertions, 45 deletions
diff --git a/tensorflow/c/c_api_function_test.cc b/tensorflow/c/c_api_function_test.cc index 4ffc9d6931..d5580b6589 100644 --- a/tensorflow/c/c_api_function_test.cc +++ b/tensorflow/c/c_api_function_test.cc @@ -1482,51 +1482,6 @@ TEST_F(CApiFunctionTest, GetOpDef) { EXPECT_EQ(op_def.name(), func_name_); EXPECT_EQ(op_def.input_arg_size(), 1); EXPECT_EQ(op_def.output_arg_size(), 1); - EXPECT_FALSE(op_def.is_stateful()); - - TF_DeleteBuffer(buffer); -} - -void DefineStatefulFunction(const char* name, TF_Function** func) { - std::unique_ptr<TF_Graph, decltype(&TF_DeleteGraph)> func_graph( - TF_NewGraph(), TF_DeleteGraph); - std::unique_ptr<TF_Status, decltype(&TF_DeleteStatus)> s(TF_NewStatus(), - TF_DeleteStatus); - - TF_Tensor* tensor_shape = Int32Tensor({37, 1}); - TF_Operation* shape = Const(tensor_shape, func_graph.get(), s.get(), "shape"); - TF_Operation* random = - RandomUniform(shape, TF_FLOAT, func_graph.get(), s.get()); - - TF_Output inputs[] = {}; - TF_Output outputs[] = {{random, 0}}; - *func = TF_GraphToFunction(func_graph.get(), name, /*append_hash=*/0, -1, - /*opers=*/nullptr, 0, inputs, 1, outputs, - /*output_names=*/nullptr, - /*opts=*/nullptr, "", s.get()); - ASSERT_EQ(TF_OK, TF_GetCode(s.get())) << TF_Message(s.get()); - ASSERT_NE(*func, nullptr); - TF_DeleteTensor(tensor_shape); -} - -TEST_F(CApiFunctionTest, StatefulOpDef) { - DefineStatefulFunction(func_name_, &func_); - TF_GraphCopyFunction(host_graph_, func_, nullptr, s_); - ASSERT_EQ(TF_OK, TF_GetCode(s_)) << TF_Message(s_); - - // Test we can retrieve function OpDef from graph - TF_Buffer* buffer = TF_NewBuffer(); - TF_GraphGetOpDef(host_graph_, func_name_, buffer, s_); - ASSERT_EQ(TF_OK, TF_GetCode(s_)) << TF_Message(s_); - - // Sanity check returned OpDef - string data(static_cast<const char*>(buffer->data), buffer->length); - OpDef op_def; - op_def.ParseFromString(data); - EXPECT_EQ(op_def.name(), func_name_); - EXPECT_EQ(op_def.input_arg_size(), 0); - EXPECT_EQ(op_def.output_arg_size(), 1); - EXPECT_TRUE(op_def.is_stateful()); TF_DeleteBuffer(buffer); } |