Home
last modified time | relevance | path

Searched refs:ContainsNodeWithOp (Results 1 – 11 of 11) sorted by relevance

/external/tensorflow/tensorflow/core/grappler/optimizers/data/
Dmap_and_filter_fusion_test.cc56 EXPECT_TRUE(graph_utils::ContainsNodeWithOp("MapDataset", output)); in TEST()
59 graph_utils::ContainsNodeWithOp("FilterByLastComponentDataset", output)); in TEST()
87 EXPECT_TRUE(graph_utils::ContainsNodeWithOp("ParallelMapDataset", output)) in TEST()
93 graph_utils::ContainsNodeWithOp("FilterByLastComponentDataset", output)) in TEST()
120 ASSERT_TRUE(graph_utils::ContainsNodeWithOp("MapDataset", output)); in TEST()
122 graph_utils::ContainsNodeWithOp("FilterByLastComponentDataset", output)); in TEST()
123 ASSERT_TRUE(graph_utils::ContainsNodeWithOp("CacheDataset", output)); in TEST()
169 ASSERT_TRUE(graph_utils::ContainsNodeWithOp("ParallelMapDataset", output)); in TEST()
171 graph_utils::ContainsNodeWithOp("FilterByLastComponentDataset", output)); in TEST()
172 ASSERT_TRUE(graph_utils::ContainsNodeWithOp("CacheDataset", output)); in TEST()
Dmake_numa_aware_test.cc60 EXPECT_FALSE(graph_utils::ContainsNodeWithOp("ExperimentalMapAndBatchDataset", in TEST()
62 EXPECT_TRUE(graph_utils::ContainsNodeWithOp( in TEST()
95 EXPECT_FALSE(graph_utils::ContainsNodeWithOp("ExperimentalMapAndBatchDataset", in TEST()
97 EXPECT_TRUE(graph_utils::ContainsNodeWithOp( in TEST()
99 EXPECT_TRUE(graph_utils::ContainsNodeWithOp("CacheDataset", output)); in TEST()
Dfilter_fusion_test.cc51 EXPECT_TRUE(graph_utils::ContainsNodeWithOp("FilterDataset", output)); in TEST()
76 EXPECT_TRUE(graph_utils::ContainsNodeWithOp("FilterDataset", output)); in TEST()
Dmap_parallelization_test.cc52 EXPECT_TRUE(graph_utils::ContainsNodeWithOp("ParallelMapDataset", output)); in TEST()
78 EXPECT_TRUE(graph_utils::ContainsNodeWithOp("ParallelMapDataset", output)); in TEST()
Dmap_fusion_test.cc51 EXPECT_TRUE(graph_utils::ContainsNodeWithOp("MapDataset", output)); in TEST()
76 EXPECT_TRUE(graph_utils::ContainsNodeWithOp("MapDataset", output)); in TEST()
Dgraph_utils_test.cc126 TEST(GraphUtilsTest, ContainsNodeWithOp) { in TEST() argument
129 EXPECT_TRUE(!ContainsNodeWithOp("OpA", *graph.graph())); in TEST()
132 EXPECT_TRUE(ContainsNodeWithOp("OpA", *graph.graph())); in TEST()
135 EXPECT_TRUE(!ContainsNodeWithOp("OpA", *graph.graph())); in TEST()
Dmap_and_batch_fusion_test.cc87 EXPECT_TRUE(graph_utils::ContainsNodeWithOp("ExperimentalMapAndBatchDataset", in TEST()
173 EXPECT_TRUE(graph_utils::ContainsNodeWithOp("ExperimentalMapAndBatchDataset", in TEST()
257 EXPECT_TRUE(graph_utils::ContainsNodeWithOp("ExperimentalMapAndBatchDataset", in TEST()
Dlatency_all_edges_test.cc60 EXPECT_TRUE(graph_utils::ContainsNodeWithOp("ExperimentalLatencyStatsDataset", in TEST()
Dgraph_utils.h93 bool ContainsNodeWithOp(StringPiece op, const GraphDef& graph);
Dshuffle_and_repeat_fusion_test.cc79 graph_utils::ContainsNodeWithOp("ShuffleAndRepeatDataset", output)); in TEST()
Dgraph_utils.cc199 bool ContainsNodeWithOp(StringPiece op, const GraphDef& graph) { in ContainsNodeWithOp() function