diff options
author | A. Unique TensorFlower <gardener@tensorflow.org> | 2016-07-13 16:16:20 -0800 |
---|---|---|
committer | TensorFlower Gardener <gardener@tensorflow.org> | 2016-07-13 17:31:25 -0700 |
commit | 078d4a2828360728e4424b3ef057808d9185a87e (patch) | |
tree | 35c008f0a70a2ffc101e5e70375c600cba71e67a /tensorflow/core/client/tensor_c_api_test.cc | |
parent | 5d516c83722314fbeb138a33009ef96854fd35e0 (diff) |
Add TF_NodeOutputListLength() and TF_NodeInputListLength() to C API
(executing a TODO).
Change: 127378577
Diffstat (limited to 'tensorflow/core/client/tensor_c_api_test.cc')
-rw-r--r-- | tensorflow/core/client/tensor_c_api_test.cc | 9 |
1 files changed, 9 insertions, 0 deletions
diff --git a/tensorflow/core/client/tensor_c_api_test.cc b/tensorflow/core/client/tensor_c_api_test.cc index 52fe0ff1e8..f89c577d75 100644 --- a/tensorflow/core/client/tensor_c_api_test.cc +++ b/tensorflow/core/client/tensor_c_api_test.cc @@ -344,11 +344,16 @@ TEST(CAPI, Graph) { EXPECT_EQ(string(""), string(TF_NodeDevice(feed))); EXPECT_EQ(1, TF_NodeNumOutputs(feed)); EXPECT_EQ(TF_INT32, TF_NodeOutputType(TF_Port{feed, 0})); + EXPECT_EQ(1, TF_NodeOutputListLength(feed, "output", s)); + ASSERT_EQ(TF_OK, TF_GetCode(s)) << TF_Message(s); EXPECT_EQ(0, TF_NodeNumInputs(feed)); EXPECT_EQ(0, TF_NodeOutputNumConsumers(TF_Port{feed, 0})); EXPECT_EQ(0, TF_NodeNumControlInputs(feed)); EXPECT_EQ(0, TF_NodeNumControlOutputs(feed)); + EXPECT_EQ(-1, TF_NodeOutputListLength(feed, "bogus", s)); + EXPECT_EQ(TF_INVALID_ARGUMENT, TF_GetCode(s)); + // Make a constant node with the scalar "3". TF_Node* three = ScalarConst(3, graph, s); ASSERT_EQ(TF_OK, TF_GetCode(s)) << TF_Message(s); @@ -363,7 +368,11 @@ TEST(CAPI, Graph) { EXPECT_EQ(string(""), string(TF_NodeDevice(add))); EXPECT_EQ(1, TF_NodeNumOutputs(add)); EXPECT_EQ(TF_INT32, TF_NodeOutputType(TF_Port{add, 0})); + EXPECT_EQ(1, TF_NodeOutputListLength(add, "sum", s)); + ASSERT_EQ(TF_OK, TF_GetCode(s)) << TF_Message(s); EXPECT_EQ(2, TF_NodeNumInputs(add)); + EXPECT_EQ(2, TF_NodeInputListLength(add, "inputs", s)); + ASSERT_EQ(TF_OK, TF_GetCode(s)) << TF_Message(s); EXPECT_EQ(TF_INT32, TF_NodeInputType(TF_Port{add, 0})); EXPECT_EQ(TF_INT32, TF_NodeInputType(TF_Port{add, 1})); TF_Port add_in_0 = TF_NodeInput(TF_Port{add, 0}); |