Searched refs:_modify_graph_op_device (Results 1 – 1 of 1) sorted by relevance
490 def _modify_graph_op_device( function512 _modify_graph_op_device(submod, new_device)515 _modify_graph_op_device(submod.graph_module, new_device)1131 _modify_graph_op_device(stage_module, device)