| /external/tensorflow/tensorflow/core/grappler/optimizers/data/ |
| D | map_and_filter_fusion_test.cc | 58 EXPECT_TRUE(graph_utils::ContainsNodeWithOp("FilterDataset", output)); in TEST()
86 ASSERT_TRUE(graph_utils::ContainsNodeWithOp("ParallelMapDataset", output)); in TEST()
87 EXPECT_TRUE(graph_utils::ContainsNodeWithOp("MapDataset", output)); in TEST()
88 EXPECT_TRUE(graph_utils::ContainsNodeWithOp("FilterDataset", output)); in TEST()
122 EXPECT_TRUE(graph_utils::ContainsNodeWithOp("FilterDataset", output)); in TEST()
123 EXPECT_TRUE(graph_utils::ContainsNodeWithOp("CacheDataset", output)); in TEST()
153 EXPECT_TRUE(graph_utils::ContainsNodeWithOp("FilterDataset", output)); in TEST()
154 ASSERT_TRUE(graph_utils::ContainsNodeWithOp("ParallelMapDataset", output)); in TEST()
155 EXPECT_TRUE(graph_utils::ContainsNodeWithOp("CacheDataset", output)); in TEST()
|
| D | disable_intra_op_parallelism_test.cc | 71 EXPECT_TRUE(graph_utils::ContainsNodeWithOp(op, item.graph)); in TEST_P()
79 EXPECT_TRUE(graph_utils::ContainsNodeWithOp(op, output)); in TEST_P()
114 EXPECT_FALSE(graph_utils::ContainsNodeWithOp("MaxIntraOpParallelismDataset", in TEST_P()
123 graph_utils::ContainsNodeWithOp("MaxIntraOpParallelismDataset", output)); in TEST_P()
129 EXPECT_FALSE(graph_utils::ContainsNodeWithOp("MaxIntraOpParallelismDataset", in TEST_P()
137 graph_utils::ContainsNodeWithOp("MaxIntraOpParallelismDataset", output)); in TEST_P()
170 EXPECT_FALSE(graph_utils::ContainsNodeWithOp("MaxIntraOpParallelismDataset", in TEST()
181 graph_utils::ContainsNodeWithOp("MaxIntraOpParallelismDataset", output)); in TEST()
|
| D | map_parallelization_test.cc | 71 EXPECT_EQ(graph_utils::ContainsNodeWithOp("ParallelMapDatasetV2", output), in TEST_P()
102 EXPECT_EQ(graph_utils::ContainsNodeWithOp("ParallelMapDatasetV2", output), in TEST_P()
134 EXPECT_TRUE(graph_utils::ContainsNodeWithOp("ParallelMapDatasetV2", output)); in TEST()
|
| D | batch_parallelization_test.cc | 70 EXPECT_EQ(graph_utils::ContainsNodeWithOp("ParallelBatchDataset", output), in TEST_P()
102 EXPECT_EQ(graph_utils::ContainsNodeWithOp("ParallelBatchDataset", output), in TEST_P()
137 EXPECT_TRUE(graph_utils::ContainsNodeWithOp("ParallelBatchDataset", output)); in TEST()
|
| D | filter_with_random_uniform_fusion_test.cc | 50 EXPECT_TRUE(graph_utils::ContainsNodeWithOp("SamplingDataset", output)); in TEST()
72 EXPECT_FALSE(graph_utils::ContainsNodeWithOp("SamplingDataset", 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_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 | 184 TEST(GraphUtilsTest, ContainsNodeWithOp) { in TEST() argument
187 EXPECT_TRUE(!ContainsNodeWithOp("OpA", *graph.graph())); in TEST()
190 EXPECT_TRUE(ContainsNodeWithOp("OpA", *graph.graph())); in TEST()
193 EXPECT_TRUE(!ContainsNodeWithOp("OpA", *graph.graph())); in TEST()
|
| D | slack_test.cc | 75 ASSERT_TRUE(graph_utils::ContainsNodeWithOp("PrefetchDataset", output)); in TEST_P()
167 ASSERT_TRUE(graph_utils::ContainsNodeWithOp("PrefetchDataset", output)); in TEST()
|
| D | shuffle_and_repeat_fusion_test.cc | 84 graph_utils::ContainsNodeWithOp("ShuffleAndRepeatDataset", output)); in TEST()
150 graph_utils::ContainsNodeWithOp("ShuffleAndRepeatDatasetV2", output)); in TEST()
220 graph_utils::ContainsNodeWithOp("ShuffleAndRepeatDatasetV2", output)); in TEST()
|
| D | map_and_batch_fusion_test.cc | 87 EXPECT_TRUE(graph_utils::ContainsNodeWithOp("MapAndBatchDataset", output)); in TEST()
172 EXPECT_TRUE(graph_utils::ContainsNodeWithOp("MapAndBatchDataset", output)); in TEST()
255 EXPECT_TRUE(graph_utils::ContainsNodeWithOp("MapAndBatchDataset", output)); in TEST()
340 EXPECT_TRUE(graph_utils::ContainsNodeWithOp("MapAndBatchDataset", output)); in TEST()
|
| D | autotune_buffer_sizes_test.cc | 105 EXPECT_TRUE(graph_utils::ContainsNodeWithOp("PrefetchDataset", output)); in TEST_P()
147 EXPECT_EQ(graph_utils::ContainsNodeWithOp("PrefetchDataset", output), in TEST_P()
|
| D | latency_all_edges_test.cc | 72 EXPECT_TRUE(graph_utils::ContainsNodeWithOp("LatencyStatsDataset", output)); in TEST_P()
|
| D | graph_utils.h | 109 bool ContainsNodeWithOp(StringPiece op, const GraphDef& graph);
|
| D | graph_utils.cc | 241 bool ContainsNodeWithOp(StringPiece op, const GraphDef& graph) { in ContainsNodeWithOp() function
|