Searched refs:UpdateFanin (Results 1 – 4 of 4) sorted by relevance
/external/tensorflow/tensorflow/core/grappler/optimizers/data/ |
D | autotune_buffer_sizes.cc | 78 TF_RETURN_IF_ERROR(graph.UpdateFanin(node.name(), in OptimizeAndCollectStats()
|
/external/tensorflow/tensorflow/core/grappler/ |
D | mutable_graph_view.h | 219 Status UpdateFanin(absl::string_view node_name, const TensorId& from_fanin,
|
D | mutable_graph_view_test.cc | 1555 Status s = graph.UpdateFanin(node_name, from_fanin, to_fanin); in TestUpdateFanin() 1572 TEST(MutableGraphViewTest, UpdateFanin) { in TEST() argument 1690 Status s = graph.UpdateFanin("c", fanin, {"b", Graph::kControlSlot}); in TestUpdateFaninFromFaninToNodeAsSwitchControl() 2241 TF_EXPECT_OK(graph.UpdateFanin("c", {"a", 1}, {"b", 2})); in TEST() 2259 TF_EXPECT_OK(graph.UpdateFanin("d", {"b", Graph::kControlSlot}, in TEST() 2263 TF_EXPECT_OK(graph.UpdateFanin("e", {"b", 0}, {"c", 3})); in TEST() 2266 TF_EXPECT_OK(graph.UpdateFanin("e", {"c", 3}, {"c", Graph::kControlSlot})); in TEST() 2378 TF_EXPECT_OK(graph.UpdateFanin("c", {"a", 2}, {"b", 3})); in TEST()
|
D | mutable_graph_view.cc | 1235 Status MutableGraphView::UpdateFanin(absl::string_view node_name, in UpdateFanin() function in tensorflow::grappler::MutableGraphView
|