Searched refs:ContainsNodeWithOp (Results 1 – 11 of 11) sorted by relevance
/external/tensorflow/tensorflow/core/grappler/optimizers/data/ |
D | map_and_filter_fusion_test.cc | 56 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()
|
D | make_numa_aware_test.cc | 60 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()
|
D | filter_fusion_test.cc | 51 EXPECT_TRUE(graph_utils::ContainsNodeWithOp("FilterDataset", output)); in TEST() 76 EXPECT_TRUE(graph_utils::ContainsNodeWithOp("FilterDataset", output)); in TEST()
|
D | map_parallelization_test.cc | 52 EXPECT_TRUE(graph_utils::ContainsNodeWithOp("ParallelMapDataset", output)); in TEST() 78 EXPECT_TRUE(graph_utils::ContainsNodeWithOp("ParallelMapDataset", output)); in TEST()
|
D | map_fusion_test.cc | 51 EXPECT_TRUE(graph_utils::ContainsNodeWithOp("MapDataset", output)); in TEST() 76 EXPECT_TRUE(graph_utils::ContainsNodeWithOp("MapDataset", output)); in TEST()
|
D | graph_utils_test.cc | 126 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()
|
D | map_and_batch_fusion_test.cc | 87 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()
|
D | latency_all_edges_test.cc | 60 EXPECT_TRUE(graph_utils::ContainsNodeWithOp("ExperimentalLatencyStatsDataset", in TEST()
|
D | graph_utils.h | 93 bool ContainsNodeWithOp(StringPiece op, const GraphDef& graph);
|
D | shuffle_and_repeat_fusion_test.cc | 79 graph_utils::ContainsNodeWithOp("ShuffleAndRepeatDataset", output)); in TEST()
|
D | graph_utils.cc | 199 bool ContainsNodeWithOp(StringPiece op, const GraphDef& graph) { in ContainsNodeWithOp() function
|