Searched refs:move_graph_nodes_to_cuda (Results 1 – 1 of 1) sorted by relevance
255 def move_graph_nodes_to_cuda(self, graph) -> List[int]: member in AutogradCompilerInstance313 runtime_inputs_to_move = self.move_graph_nodes_to_cuda(self.fx_tracer.graph)