Searched refs:push_node (Results 1 – 1 of 1) sorted by relevance
532 NodeDef* push_node = optimized_graph->mutable_node(push_node_idx); in RemoveStackOps() local534 << push_node->DebugString(); in RemoveStackOps()535 if (push_node->attr().count("swap_memory") != 0) { in RemoveStackOps()536 push_node->mutable_attr()->erase("swap_memory"); in RemoveStackOps()538 push_node->set_op("Identity"); in RemoveStackOps()539 push_node->mutable_input()->SwapElements(0, 1); in RemoveStackOps()541 push_node->input(1), optimized_graph, &node_map); in RemoveStackOps()542 push_node->set_input(1, ctrl_dep); in RemoveStackOps()543 VLOG(1) << "After converting: " << push_node->DebugString(); in RemoveStackOps()