Searched refs:FindGraphNodeWithName (Results 1 – 7 of 7) sorted by relevance
| /external/tensorflow/tensorflow/core/grappler/optimizers/data/ |
| D | make_sloppy_test.cc | 60 int index = graph_utils::FindGraphNodeWithName("interleave", output); in TEST()
85 int index = graph_utils::FindGraphNodeWithName("map", output); in TEST()
108 int index = graph_utils::FindGraphNodeWithName("parse_example", output); in TEST()
|
| D | graph_utils_test.cc | 138 TEST(GraphUtilsTest, FindGraphNodeWithName) { in TEST() argument
141 EXPECT_EQ(FindGraphNodeWithName("A", *graph.graph()), -1); in TEST()
144 EXPECT_NE(FindGraphNodeWithName("A", *graph.graph()), -1); in TEST()
147 EXPECT_EQ(FindGraphNodeWithName("A", *graph.graph()), -1); in TEST()
172 EXPECT_EQ(FindGraphNodeWithName("A2", *graph.graph()), 1); in TEST()
|
| D | map_and_batch_fusion_test.cc | 96 graph_utils::FindGraphNodeWithName(map_and_batch_node.input(3), output)); in TEST()
100 graph_utils::FindGraphNodeWithName(map_and_batch_node.input(4), output)); in TEST()
182 graph_utils::FindGraphNodeWithName(map_and_batch_node.input(3), output)); in TEST()
266 graph_utils::FindGraphNodeWithName(map_and_batch_node.input(3), output)); in TEST()
270 graph_utils::FindGraphNodeWithName(map_and_batch_node.input(4), output)); in TEST()
|
| D | noop_elimination_test.cc | 143 graph_utils::FindGraphNodeWithName(cache_node->name(), output)); in TEST_P()
204 graph_utils::FindGraphNodeWithName(cache_node->name(), output)); in TEST_P()
|
| D | map_parallelization.cc | 40 int index = graph_utils::FindGraphNodeWithName(name, *graph->graph()); in MakeParallelMap()
|
| D | graph_utils.h | 97 int FindGraphNodeWithName(StringPiece name, const GraphDef& graph);
|
| D | graph_utils.cc | 196 return FindGraphNodeWithName(name, graph) != -1; in ContainsGraphNodeWithName()
212 int FindGraphNodeWithName(StringPiece name, const GraphDef& graph) { in FindGraphNodeWithName() function
|