Searched refs:new_mul_node (Results 1 – 2 of 2) sorted by relevance
2443 NodeDef* new_mul_node = AddEmptyNode(optimized_mul_name); in TrySimplify() local2444 new_mul_node->set_op("Mul"); in TrySimplify()2445 new_mul_node->set_device(node->device()); in TrySimplify()2446 SetDataTypeToAttr(type, "T", new_mul_node); in TrySimplify()2447 new_mul_node->add_input(new_const_node->name()); in TrySimplify()2448 ctx().node_map->AddOutput(new_const_node->name(), new_mul_node->name()); in TrySimplify()2449 new_mul_node->add_input(node->input(0)); in TrySimplify()2450 ctx().node_map->AddOutput(node->input(0), new_mul_node->name()); in TrySimplify()2452 ForwardControlDependencies(new_mul_node, {node}); in TrySimplify()2453 *simplified_node_name = new_mul_node->name(); in TrySimplify()
575 const NodeDef* new_mul_node = node_map.GetNode(HoistMulName("add")); in TEST_F() local576 ASSERT_NE(new_mul_node, nullptr) << "Hoisted Mul node not found"; in TEST_F()577 EXPECT_EQ("x", new_mul_node->input(0)); in TEST_F()578 EXPECT_EQ(new_add_node->name(), new_mul_node->input(1)); in TEST_F()