Searched refs:inner_square_node (Results 1 – 1 of 1) sorted by relevance
2613 NodeDef* inner_square_node = ctx().node_map->GetNode(inner_square_name); in TrySimplify() local2614 if (inner_square_node == nullptr) { in TrySimplify()2615 inner_square_node = AddCopyNode(inner_square_name, node); in TrySimplify()2616 inner_square_node->set_op("Square"); in TrySimplify()2617 inner_square_node->mutable_input()->RemoveLast(); in TrySimplify()2619 ctx().node_map->AddOutput(x->name(), inner_square_node->name()); in TrySimplify()2622 node->set_input(1, inner_square_node->name()); in TrySimplify()2626 AddToOptimizationQueue(inner_square_node); in TrySimplify()