| 57 | }; |
| 58 | |
| 59 | TEST_F(TFProfTensorTest, Basics) { |
| 60 | Options opts(3, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, -1, "name", {"VariableV2"}, |
| 61 | {".*"}, {""}, {".*"}, {""}, false, |
| 62 | {"tensor_value"}, // show the tensor value. |
| 63 | "", {}); |
| 64 | const GraphNodeProto& root = tf_stats_->ShowGraphNode("scope", opts); |
| 65 | |
| 66 | EXPECT_EQ(root.children(0).name(), "DW"); |
| 67 | EXPECT_GT(root.children(0).tensor_value().value_double_size(), 10); |
| 68 | EXPECT_EQ(root.children(1).name(), "DW2"); |
| 69 | EXPECT_GT(root.children(1).tensor_value().value_double_size(), 10); |
| 70 | EXPECT_EQ(root.children(2).name(), "ScalarW"); |
| 71 | EXPECT_EQ(root.children(2).tensor_value().value_double_size(), 1); |
| 72 | } |
| 73 | |
| 74 | } // namespace tfprof |
| 75 | } // namespace tensorflow |