Searched refs:FindGraphNodeWithOp (Results 1 – 9 of 9) sorted by relevance
| /external/tensorflow/tensorflow/core/grappler/optimizers/data/ |
| D | map_and_filter_fusion_test.cc | 90 graph_utils::FindGraphNodeWithOp("ParallelMapDataset", output)); in TEST()
125 int map_id = graph_utils::FindGraphNodeWithOp("MapDataset", output); in TEST()
131 graph_utils::FindGraphNodeWithOp("FilterByLastComponentDataset", output); in TEST()
136 int cache_id = graph_utils::FindGraphNodeWithOp("CacheDataset", output); in TEST()
174 int map_id = graph_utils::FindGraphNodeWithOp("ParallelMapDataset", output); in TEST()
181 graph_utils::FindGraphNodeWithOp("FilterByLastComponentDataset", output); in TEST()
186 int cache_id = graph_utils::FindGraphNodeWithOp("CacheDataset", output); in TEST()
|
| D | hoist_random_uniform_test.cc | 54 const int new_map_id = graph_utils::FindGraphNodeWithOp("MapDataset", output); in TEST()
56 graph_utils::FindGraphNodeWithOp("ZipDataset", output); in TEST()
58 graph_utils::FindGraphNodeWithOp("ExperimentalRandomDataset", output); in TEST()
60 graph_utils::FindGraphNodeWithOp("BatchDatasetV2", output); in TEST()
|
| D | graph_utils_test.cc | 160 TEST(GraphUtilsTest, FindGraphNodeWithOp) { in TEST() argument
163 EXPECT_EQ(FindGraphNodeWithOp("OpA", *graph.graph()), -1); in TEST()
168 EXPECT_EQ(FindGraphNodeWithOp("OpA", *graph.graph()), 0); in TEST()
171 EXPECT_EQ(FindGraphNodeWithOp("OpB", *graph.graph()), -1); in TEST()
178 EXPECT_EQ(FindGraphNodeWithOp("OpA", *graph.graph()), -1); in TEST()
|
| D | make_numa_aware_test.cc | 101 int numa_map_and_batch_component_id = graph_utils::FindGraphNodeWithOp( in TEST()
107 int cache_id = graph_utils::FindGraphNodeWithOp("CacheDataset", output); in TEST()
|
| D | map_vectorization_test.cc | 168 output.node(graph_utils::FindGraphNodeWithOp(map_op, output)); in CheckNotVectorized()
170 output.node(graph_utils::FindGraphNodeWithOp(batch_op, output)); in CheckNotVectorized()
208 output.node(graph_utils::FindGraphNodeWithOp(kChooseFastestOp, output)); in CheckVectorized()
371 output.node(graph_utils::FindGraphNodeWithOp(kRangeOp, output)); in TEST_P()
|
| D | map_and_batch_fusion_test.cc | 89 NodeDef map_and_batch_node = output.node(graph_utils::FindGraphNodeWithOp( in TEST()
175 NodeDef map_and_batch_node = output.node(graph_utils::FindGraphNodeWithOp( in TEST()
259 NodeDef map_and_batch_node = output.node(graph_utils::FindGraphNodeWithOp( in TEST()
|
| D | graph_utils.h | 106 int FindGraphNodeWithOp(StringPiece op, const GraphDef& graph);
|
| D | graph_utils.cc | 200 return FindGraphNodeWithOp(op, graph) != -1; in ContainsNodeWithOp()
218 int FindGraphNodeWithOp(StringPiece op, const GraphDef& graph) { in FindGraphNodeWithOp() function
|
| D | shuffle_and_repeat_fusion_test.cc | 81 graph_utils::FindGraphNodeWithOp("ShuffleAndRepeatDataset", output)); in TEST()
|