Searched refs:FindAllGraphNodesWithOp (Results 1 – 5 of 5) sorted by relevance
| /external/tensorflow/tensorflow/core/grappler/optimizers/data/ |
| D | graph_utils_test.cc | 175 TEST(GraphUtilsTest, FindAllGraphNodesWithOp) { in TEST() argument
184 FindAllGraphNodesWithOp("OpA", *graph.graph()); in TEST()
191 FindAllGraphNodesWithOp("OpA", *graph.graph()); in TEST()
|
| D | map_vectorization_test.cc | 165 ASSERT_EQ(graph_utils::FindAllGraphNodesWithOp(map_op, output).size(), 1); in CheckNotVectorized()
166 ASSERT_EQ(graph_utils::FindAllGraphNodesWithOp(batch_op, output).size(), 1); in CheckNotVectorized()
206 graph_utils::FindAllGraphNodesWithOp(kChooseFastestOp, output).size(), 1); in CheckVectorized()
365 graph_utils::FindAllGraphNodesWithOp(kChooseFastestOp, output); in TEST_P()
|
| D | latency_all_edges_test.cc | 62 std::vector<int> latency_node_indices = graph_utils::FindAllGraphNodesWithOp( in TEST()
|
| D | graph_utils.h | 117 std::vector<int> FindAllGraphNodesWithOp(const string& op,
|
| D | graph_utils.cc | 223 std::vector<int> FindAllGraphNodesWithOp(const string& op, in FindAllGraphNodesWithOp() function
|