Searched refs:swap_out_node (Results 1 – 1 of 1) sorted by relevance
744 NodeDef* swap_out_node = graph->add_node(); in BuildSwapPair() local745 swap_out_node->set_name(swap_out_name); in BuildSwapPair()746 swap_out_node->set_op("_CopyFromGpuToHost"); in BuildSwapPair()752 *swap_in_node->add_input() = swap_out_node->name(); in BuildSwapPair()755 swap_out_node->set_device(node->device()); in BuildSwapPair()758 (*swap_out_node->mutable_attr())["_class"].mutable_list()->add_s(coloc_group); in BuildSwapPair()763 (*swap_out_node->mutable_attr())["T"].set_type(input_type); in BuildSwapPair()764 *swap_pair = std::make_pair(swap_out_node, swap_in_node); in BuildSwapPair()