Searched refs:CountOpNodes (Results 1 – 4 of 4) sorted by relevance
| /external/tensorflow/tensorflow/core/grappler/utils/ |
| D | grappler_test_test.cc | 68 TEST_F(GrapplerTestTest, CountOpNodes) { in TEST_F() argument
92 EXPECT_EQ(2, CountOpNodes(graph, "Add")); in TEST_F()
93 EXPECT_EQ(2, CountOpNodes(graph, "Mul")); in TEST_F()
94 EXPECT_EQ(1, CountOpNodes(graph, "AddN")); in TEST_F()
95 EXPECT_EQ(0, CountOpNodes(graph, "Transpose")); in TEST_F()
|
| D | grappler_test.h | 78 int CountOpNodes(const GraphDef& graph, const string& op);
|
| D | grappler_test.cc | 184 int GrapplerTest::CountOpNodes(const GraphDef& graph, const string& op) { in CountOpNodes() function in tensorflow::grappler::GrapplerTest
|
| /external/tensorflow/tensorflow/core/grappler/optimizers/ |
| D | arithmetic_optimizer_test.cc | 921 EXPECT_EQ(0, CountOpNodes(output, "Reshape")); in TEST_F()
963 EXPECT_EQ(0, CountOpNodes(output, "Reshape")); in TEST_F()
993 EXPECT_EQ(1, CountOpNodes(output, "Reshape")); in TEST_F()
1023 EXPECT_EQ(0, CountOpNodes(output, "Reshape")); in TEST_F()
1051 EXPECT_EQ(1, CountOpNodes(output, "Reshape")); in TEST_F()
1074 EXPECT_EQ(1, CountOpNodes(output, "Reshape")); in TEST_F()
1108 EXPECT_EQ(1, CountOpNodes(output, "Reshape")); in TEST_F()
1780 EXPECT_EQ(1, CountOpNodes(output, "Bitcast")); in TEST_F()
1814 EXPECT_EQ(0, CountOpNodes(output, "Bitcast")); in TEST_F()
1847 EXPECT_EQ(0, CountOpNodes(output, "Cast")); in TEST_F()
|