Searched refs:DeleteNodes (Results 1 – 17 of 17) sorted by relevance
| /external/tensorflow/tensorflow/core/grappler/optimizers/data/ |
| D | graph_utils_test.cc | 170 EXPECT_TRUE(graph.DeleteNodes({"A"}).ok()); in TEST()
192 EXPECT_TRUE(graph.DeleteNodes({"A"}).ok()); in TEST()
204 EXPECT_TRUE(graph.DeleteNodes({"A"}).ok()); in TEST()
228 EXPECT_TRUE(graph.DeleteNodes({"B"}).ok()); in TEST()
247 EXPECT_TRUE(graph.DeleteNodes({"A2"}).ok()); in TEST()
262 EXPECT_TRUE(graph.DeleteNodes({node1->name()}).ok()); in TEST()
|
| D | batch_parallelization.cc | 98 TF_RETURN_IF_ERROR(graph.DeleteNodes(nodes_to_delete)); in OptimizeAndCollectStats()
|
| D | map_parallelization.cc | 100 TF_RETURN_IF_ERROR(graph.DeleteNodes(nodes_to_delete)); in OptimizeAndCollectStats()
|
| D | filter_fusion.cc | 125 TF_RETURN_IF_ERROR(graph.DeleteNodes(nodes_to_delete)); in OptimizeAndCollectStats()
|
| D | map_and_batch_fusion.cc | 143 TF_RETURN_IF_ERROR(graph.DeleteNodes(nodes_to_delete)); in OptimizeAndCollectStats()
|
| D | noop_elimination.cc | 156 TF_RETURN_IF_ERROR(graph.DeleteNodes(nodes_to_delete)); in OptimizeAndCollectStats()
|
| D | map_fusion.cc | 147 TF_RETURN_IF_ERROR(graph.DeleteNodes(nodes_to_delete)); in OptimizeAndCollectStats()
|
| D | shuffle_and_repeat_fusion.cc | 204 TF_RETURN_IF_ERROR(graph.DeleteNodes(nodes_to_delete)); in OptimizeAndCollectStats()
|
| D | filter_with_random_uniform_fusion.cc | 261 TF_RETURN_IF_ERROR(graph.DeleteNodes(nodes_to_delete)); in OptimizeAndCollectStats()
|
| D | map_and_filter_fusion.cc | 246 TF_RETURN_IF_ERROR(graph.DeleteNodes(nodes_to_delete)); in OptimizeAndCollectStats()
|
| D | hoist_random_uniform.cc | 291 TF_RETURN_IF_ERROR(graph.DeleteNodes(nodes_to_delete)); in OptimizeAndCollectStats()
|
| D | map_vectorization.cc | 619 TF_RETURN_IF_ERROR(graph.DeleteNodes(nodes_to_delete)); in OptimizeAndCollectStats()
|
| D | auto_shard.cc | 559 return graph->DeleteNodes(nodes_to_delete); in ShardByFile()
|
| /external/tensorflow/tensorflow/core/grappler/ |
| D | mutable_graph_view.h | 249 Status DeleteNodes(const absl::flat_hash_set<string>& nodes_to_delete);
|
| D | mutable_graph_view_test.cc | 2832 TEST(MutableGraphViewTest, DeleteNodes) { in TEST() argument
2844 TF_EXPECT_OK(graph.DeleteNodes({"foo_1"})); in TEST()
2874 TF_EXPECT_OK(graph.DeleteNodes({"c", "a", "b"})); in TEST()
2895 TF_EXPECT_OK(graph.DeleteNodes({"g", "h"})); in TEST()
2919 TF_EXPECT_OK(graph.DeleteNodes({"d", "e", "f", "g", "h"})); in TEST()
2938 Status s = graph.DeleteNodes({"b", "a"}); in TEST()
2972 Status s = graph.DeleteNodes({"a", "b", "c", "d", "e", "f"}); in TEST()
|
| D | utils_test.cc | 443 TEST_F(UtilsTest, DeleteNodes) { in TEST_F() argument
|
| D | mutable_graph_view.cc | 1553 Status MutableGraphView::DeleteNodes( in DeleteNodes() function in tensorflow::grappler::MutableGraphView
|