Home
last modified time | relevance | path

Searched refs:UpdateFanouts (Results 1 – 19 of 19) sorted by relevance

/external/tensorflow/tensorflow/core/grappler/optimizers/data/
Dauto_shard.cc200 graph->UpdateFanouts(add_after->name(), new_node_graph->name())); in AddShardNode()
230 graph->UpdateFanouts(add_after->name(), new_node_graph->name())); in AddShuffleDataset()
253 graph->UpdateFanouts(add_after->name(), new_node_graph->name())); in AddShuffleDatasetV2()
284 graph->UpdateFanouts(add_after->name(), new_node_graph->name())); in AddShuffleDatasetV3()
317 TF_RETURN_IF_ERROR(graph->UpdateFanouts(node.name(), node.input(0))); in RemoveShuffleDataset()
339 TF_RETURN_IF_ERROR(graph->UpdateFanouts(node.name(), node.input(0))); in RemoveShuffleDatasetV2()
366 TF_RETURN_IF_ERROR(graph->UpdateFanouts(node.name(), node.input(0))); in RemoveShuffleDatasetV3()
457 TF_RETURN_IF_ERROR(graph->UpdateFanouts(node.name(), node.input(0))); in RecursivelyHandleOp()
Dshuffle_and_repeat_fusion.cc185 TF_RETURN_IF_ERROR(graph.UpdateFanouts(repeat_node.name(), in OptimizeAndCollectStats()
189 TF_RETURN_IF_ERROR(graph.UpdateFanouts(shuffle_node.name(), in OptimizeAndCollectStats()
Ddisable_intra_op_parallelism.cc110 graph.UpdateFanouts(last_node->name(), added_node->name())); in OptimizeAndCollectStats()
Dreorder_data_discarding_ops.cc103 graph.UpdateFanouts(discard_node->name(), parent->name())); in OptimizeAndCollectStats()
Dlatency_all_edges.cc109 graph.UpdateFanouts(node.name(), latency_node_pointer->name())); in OptimizeAndCollectStats()
Dbatch_parallelization.cc93 graph.UpdateFanouts(batch_node->name(), parallel_batch->name())); in OptimizeAndCollectStats()
Dmap_parallelization.cc95 graph.UpdateFanouts(map_node->name(), parallel_map->name())); in OptimizeAndCollectStats()
Dautotune_buffer_sizes.cc132 graph.UpdateFanouts(async_dataset_node->name(), added_node->name())); in OptimizeAndCollectStats()
Dfilter_fusion.cc116 TF_RETURN_IF_ERROR(graph.UpdateFanouts(second_filter_node->name(), in OptimizeAndCollectStats()
Dmap_and_batch_fusion.cc135 graph.UpdateFanouts(batch_node.name(), new_node->name())); in OptimizeAndCollectStats()
Dmap_fusion.cc138 graph.UpdateFanouts(map_node->name(), fused_maps_node->name())); in OptimizeAndCollectStats()
Dnoop_elimination.cc150 TF_RETURN_IF_ERROR(graph.UpdateFanouts(node.name(), parent->name())); in OptimizeAndCollectStats()
Dfilter_with_random_uniform_fusion.cc254 graph.UpdateFanouts(filter_node.name(), fused_sampling->name())); in OptimizeAndCollectStats()
Dmap_vectorization.cc608 TF_RETURN_IF_ERROR(graph.UpdateFanouts(batch_node->name(), in OptimizeAndCollectStats()
613 TF_RETURN_IF_ERROR(graph.UpdateFanouts( in OptimizeAndCollectStats()
Dmap_and_filter_fusion.cc238 graph.UpdateFanouts(filter_node->name(), new_map_node->name())); in OptimizeAndCollectStats()
Dhoist_random_uniform.cc283 graph.UpdateFanouts(map_node->name(), stateless_map->name())); in OptimizeAndCollectStats()
/external/tensorflow/tensorflow/core/grappler/
Dmutable_graph_view.h133 Status UpdateFanouts(absl::string_view from_node_name,
Dmutable_graph_view_test.cc766 TF_EXPECT_OK(graph.UpdateFanouts("bar", new_bar->name())); in TEST()
796 TF_EXPECT_OK(graph.UpdateFanouts("bar_2", new_bar->name())); in TEST()
825 TF_EXPECT_OK(graph.UpdateFanouts("bar", new_bar->name())); in TEST()
845 Status s = graph.UpdateFanouts("a", "b"); in TEST()
852 s = graph.UpdateFanouts("d", "b"); in TEST()
880 TF_EXPECT_OK(graph.UpdateFanouts("c", "b")); in TEST()
Dmutable_graph_view.cc735 Status MutableGraphView::UpdateFanouts(absl::string_view from_node_name, in UpdateFanouts() function in tensorflow::grappler::MutableGraphView