Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/core/grappler/optimizers/
Darithmetic_optimizer.cc714 NodeDef* new_add_node = AddCopyNode(InnerAddNodeName(node), node); in TrySimplify() local
718 new_outer_node->set_input(0, new_add_node->name()); in TrySimplify()
722 new_outer_node->set_input(1, new_add_node->name()); in TrySimplify()
726 ctx().node_map->AddOutput(new_add_node->name(), new_outer_node->name()); in TrySimplify()
731 new_add_node->set_input(i, unique_factor_i); in TrySimplify()
732 ctx().node_map->AddOutput(unique_factor_i, new_add_node->name()); in TrySimplify()
737 *new_add_node->add_input() = ctrl_dep; in TrySimplify()
738 ctx().node_map->AddOutput(NodeName(ctrl_dep), new_add_node->name()); in TrySimplify()
742 AddToOptimizationQueue(new_add_node); in TrySimplify()
Darithmetic_optimizer_test.cc570 const NodeDef* new_add_node = node_map.GetNode(HoistAddName("add")); in TEST_F() local
571 ASSERT_NE(new_add_node, nullptr) << "Hoisted Add node not found"; in TEST_F()
572 EXPECT_EQ("y1", new_add_node->input(0)); in TEST_F()
573 EXPECT_EQ("y2", new_add_node->input(1)); in TEST_F()
578 EXPECT_EQ(new_add_node->name(), new_mul_node->input(1)); in TEST_F()
650 const NodeDef* new_add_node = node_map.GetNode(HoistAddName("add")); in TEST_F() local
651 ASSERT_TRUE(new_add_node != nullptr) << "Hoisted Add node not found"; in TEST_F()
652 EXPECT_EQ("y1", new_add_node->input(0)); in TEST_F()
653 EXPECT_EQ("y2", new_add_node->input(1)); in TEST_F()
657 EXPECT_EQ(new_add_node->name(), new_div_node->input(0)); in TEST_F()