Searched refs:new_mul_node (Results 1 – 2 of 2) sorted by relevance
2550 NodeDef* new_mul_node = AddEmptyNode(optimized_mul_name); in TrySimplify() local2551 new_mul_node->set_op("Mul"); in TrySimplify()2552 new_mul_node->set_device(node->device()); in TrySimplify()2553 SetDataTypeToAttr(type, "T", new_mul_node); in TrySimplify()2554 new_mul_node->add_input(new_const_node->name()); in TrySimplify()2555 ctx().node_map->AddOutput(new_const_node->name(), new_mul_node->name()); in TrySimplify()2556 new_mul_node->add_input(node->input(0)); in TrySimplify()2557 ctx().node_map->AddOutput(node->input(0), new_mul_node->name()); in TrySimplify()2559 ForwardControlDependencies(new_mul_node, {node}); in TrySimplify()2560 *simplified_node_name = new_mul_node->name(); in TrySimplify()
492 const NodeDef* new_mul_node = node_map.GetNode(HoistMulName("add")); in TEST_F() local493 ASSERT_NE(new_mul_node, nullptr) << "Hoisted Mul node not found"; in TEST_F()494 ASSERT_EQ(new_mul_node->input_size(), 2); in TEST_F()495 EXPECT_EQ(new_mul_node->input(0), "x"); in TEST_F()496 EXPECT_EQ(new_mul_node->input(1), new_add_node->name()); in TEST_F()