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 | 85 int CountOpNodes(const GraphDef& graph, const string& op);
|
| D | grappler_test.cc | 205 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 | 866 EXPECT_EQ(CountOpNodes(output, "Reshape"), 0); in TEST_F()
867 EXPECT_EQ(CountOpNodes(output, "BroadcastTo"), 0); in TEST_F()
910 EXPECT_EQ(CountOpNodes(output, "Reshape"), 0); in TEST_F()
940 EXPECT_EQ(CountOpNodes(output, "Reshape"), 1); in TEST_F()
970 EXPECT_EQ(CountOpNodes(output, "Reshape"), 0); in TEST_F()
998 EXPECT_EQ(CountOpNodes(output, "Reshape"), 1); in TEST_F()
1021 EXPECT_EQ(CountOpNodes(output, "Reshape"), 1); in TEST_F()
1064 EXPECT_EQ(CountOpNodes(output, "Reshape"), 1); in TEST_F()
1083 EXPECT_EQ(CountOpNodes(output, "Reshape"), 2); in TEST_F()
1813 EXPECT_EQ(CountOpNodes(output, "Bitcast"), 1); in TEST_F()
[all …]
|