Searched refs:UpdateFanouts (Results 1 – 15 of 15) sorted by relevance
| /external/tensorflow/tensorflow/core/grappler/optimizers/data/ |
| D | make_numa_aware.cc | 53 TF_RETURN_IF_ERROR(graph.UpdateFanouts(node.name(), numa_node->name())); in OptimizeAndCollectStats()
|
| D | auto_shard.cc | 137 graph->UpdateFanouts(add_after->name(), new_node_graph->name())); in AddShardNode()
163 TF_RETURN_IF_ERROR(graph->UpdateFanouts(node.name(), node.input(0))); in RemoveShuffleDataset()
|
| D | shuffle_and_repeat_fusion.cc | 89 TF_RETURN_IF_ERROR(graph.UpdateFanouts(repeat_node.name(), in OptimizeAndCollectStats()
|
| D | map_parallelization.cc | 81 graph.UpdateFanouts(map_node->name(), parallel_map->name())); in OptimizeAndCollectStats()
|
| D | noop_elimination.cc | 85 TF_RETURN_IF_ERROR(graph.UpdateFanouts(node.name(), parent->name())); in OptimizeAndCollectStats()
|
| D | latency_all_edges.cc | 92 TF_RETURN_IF_ERROR(graph.UpdateFanouts(node.name(), latency_node->name())); in OptimizeAndCollectStats()
|
| D | map_and_batch_fusion.cc | 127 graph.UpdateFanouts(batch_node.name(), new_node->name())); in OptimizeAndCollectStats()
|
| D | filter_fusion.cc | 115 TF_RETURN_IF_ERROR(graph.UpdateFanouts(second_filter_node->name(), in OptimizeAndCollectStats()
|
| D | map_fusion.cc | 137 graph.UpdateFanouts(map_node->name(), fused_maps_node->name())); in OptimizeAndCollectStats()
|
| D | map_and_filter_fusion.cc | 162 graph.UpdateFanouts(filter_node->name(), filter_by_component->name())); in OptimizeAndCollectStats()
|
| D | hoist_random_uniform.cc | 273 graph.UpdateFanouts(map_node->name(), stateless_map->name())); in OptimizeAndCollectStats()
|
| D | map_vectorization.cc | 573 TF_RETURN_IF_ERROR(graph.UpdateFanouts(batch_node->name(), in OptimizeAndCollectStats()
|
| /external/tensorflow/tensorflow/core/grappler/ |
| D | mutable_graph_view.h | 133 Status UpdateFanouts(absl::string_view from_node_name,
|
| D | mutable_graph_view_test.cc | 766 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()
|
| D | mutable_graph_view.cc | 732 Status MutableGraphView::UpdateFanouts(absl::string_view from_node_name, in UpdateFanouts() function in tensorflow::grappler::MutableGraphView
|