Home
last modified time | relevance | path

Searched refs:DeleteNodes (Results 1 – 15 of 15) sorted by relevance

/external/tensorflow/tensorflow/core/grappler/optimizers/data/
Dgraph_utils_test.cc112 EXPECT_TRUE(graph.DeleteNodes({"A"}).ok()); in TEST()
134 EXPECT_TRUE(graph.DeleteNodes({"A"}).ok()); in TEST()
146 EXPECT_TRUE(graph.DeleteNodes({"A"}).ok()); in TEST()
170 EXPECT_TRUE(graph.DeleteNodes({"B"}).ok()); in TEST()
189 EXPECT_TRUE(graph.DeleteNodes({"A2"}).ok()); in TEST()
204 EXPECT_TRUE(graph.DeleteNodes({node1->name()}).ok()); in TEST()
Dmake_numa_aware.cc57 TF_RETURN_IF_ERROR(graph.DeleteNodes(nodes_to_delete)); in OptimizeAndCollectStats()
Dshuffle_and_repeat_fusion.cc98 TF_RETURN_IF_ERROR(graph.DeleteNodes(nodes_to_delete)); in OptimizeAndCollectStats()
Dmap_parallelization.cc86 TF_RETURN_IF_ERROR(graph.DeleteNodes(nodes_to_delete)); in OptimizeAndCollectStats()
Dnoop_elimination.cc91 TF_RETURN_IF_ERROR(graph.DeleteNodes(nodes_to_delete)); in OptimizeAndCollectStats()
Dmap_and_batch_fusion.cc135 TF_RETURN_IF_ERROR(graph.DeleteNodes(nodes_to_delete)); in OptimizeAndCollectStats()
Dfilter_fusion.cc129 TF_RETURN_IF_ERROR(graph.DeleteNodes(nodes_to_delete)); in OptimizeAndCollectStats()
Dmap_fusion.cc151 TF_RETURN_IF_ERROR(graph.DeleteNodes(nodes_to_delete)); in OptimizeAndCollectStats()
Dmap_and_filter_fusion.cc172 TF_RETURN_IF_ERROR(graph.DeleteNodes(nodes_to_delete)); in OptimizeAndCollectStats()
Dauto_shard.cc248 TF_RETURN_IF_ERROR(graph.DeleteNodes(nodes_to_delete)); in OptimizeGraph()
Dhoist_random_uniform.cc281 TF_RETURN_IF_ERROR(graph.DeleteNodes(nodes_to_delete)); in OptimizeAndCollectStats()
/external/tensorflow/tensorflow/core/grappler/
Dmutable_graph_view.h249 Status DeleteNodes(const absl::flat_hash_set<string>& nodes_to_delete);
Dmutable_graph_view_test.cc2832 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()
Dutils_test.cc389 TEST_F(UtilsTest, DeleteNodes) { in TEST_F() argument
Dmutable_graph_view.cc1550 Status MutableGraphView::DeleteNodes( in DeleteNodes() function in tensorflow::grappler::MutableGraphView