Searched refs:new_square_node (Results 1 – 1 of 1) sorted by relevance
2445 NodeDef* new_square_node = AddCopyNode(optimized_node_name, node); in TrySimplify() local2446 new_square_node->set_op("Square"); in TrySimplify()2447 for (int i = 1; i < new_square_node->input_size(); ++i) { in TrySimplify()2448 new_square_node->set_input(i - 1, new_square_node->input(i)); in TrySimplify()2450 new_square_node->mutable_input()->RemoveLast(); in TrySimplify()2451 for (const string& input : new_square_node->input()) { in TrySimplify()2452 ctx().node_map->AddOutput(NodeName(input), new_square_node->name()); in TrySimplify()2454 *simplified_node_name = new_square_node->name(); in TrySimplify()