Searched refs:new_square_node (Results 1 – 1 of 1) sorted by relevance
2338 NodeDef* new_square_node = AddCopyNode(optimized_node_name, node); in TrySimplify() local2339 new_square_node->set_op("Square"); in TrySimplify()2340 for (int i = 1; i < new_square_node->input_size(); ++i) { in TrySimplify()2341 new_square_node->set_input(i - 1, new_square_node->input(i)); in TrySimplify()2343 new_square_node->mutable_input()->RemoveLast(); in TrySimplify()2344 for (const string& input : new_square_node->input()) { in TrySimplify()2345 ctx().node_map->AddOutput(NodeName(input), new_square_node->name()); in TrySimplify()2347 *simplified_node_name = new_square_node->name(); in TrySimplify()