Home
last modified time | relevance | path

Searched refs:new_add_node (Results 1 – 2 of 2) sorted by relevance

/aosp_15_r20/external/tensorflow/tensorflow/core/grappler/optimizers/
H A Darithmetic_optimizer.cc802 NodeDef* new_add_node = AddCopyNode(InnerAddNodeName(node), node); in TrySimplify() local
806 new_outer_node->set_input(0, new_add_node->name()); in TrySimplify()
810 new_outer_node->set_input(1, new_add_node->name()); in TrySimplify()
814 ctx().node_map->AddOutput(new_add_node->name(), new_outer_node->name()); in TrySimplify()
819 new_add_node->set_input(i, unique_factor_i); in TrySimplify()
820 ctx().node_map->AddOutput(unique_factor_i, new_add_node->name()); in TrySimplify()
825 *new_add_node->add_input() = ctrl_dep; in TrySimplify()
826 ctx().node_map->AddOutput(NodeName(ctrl_dep), new_add_node->name()); in TrySimplify()
830 AddToOptimizationQueue(new_add_node); in TrySimplify()
H A Darithmetic_optimizer_test.cc914 const NodeDef* new_add_node = node_map.GetNode(HoistAddName("add")); in TEST_F() local
915 ASSERT_NE(new_add_node, nullptr) << "Hoisted Add node not found"; in TEST_F()
916 ASSERT_EQ(new_add_node->input_size(), 2); in TEST_F()
917 EXPECT_EQ(new_add_node->input(0), "y1"); in TEST_F()
918 EXPECT_EQ(new_add_node->input(1), "y2"); in TEST_F()
924 EXPECT_EQ(new_mul_node->input(1), new_add_node->name()); in TEST_F()
997 const NodeDef* new_add_node = node_map.GetNode(HoistAddName("add")); in TEST_F() local
998 ASSERT_TRUE(new_add_node != nullptr) << "Hoisted Add node not found"; in TEST_F()
999 ASSERT_EQ(new_add_node->input_size(), 2); in TEST_F()
1000 EXPECT_EQ(new_add_node->input(0), "y1"); in TEST_F()
[all …]