Searched refs:swap_in_node (Results 1 – 1 of 1) sorted by relevance
749 NodeDef* swap_in_node = graph->add_node(); in BuildSwapPair() local750 swap_in_node->set_name(swap_in_name); in BuildSwapPair()751 swap_in_node->set_op("_CopyFromHostToGpu"); in BuildSwapPair()752 *swap_in_node->add_input() = swap_out_node->name(); in BuildSwapPair()756 swap_in_node->set_device(node->device()); in BuildSwapPair()759 (*swap_in_node->mutable_attr())["_class"].mutable_list()->add_s(coloc_group); in BuildSwapPair()762 (*swap_in_node->mutable_attr())["T"].set_type(input_type); in BuildSwapPair()764 *swap_pair = std::make_pair(swap_out_node, swap_in_node); in BuildSwapPair()