Home
last modified time | relevance | path

Searched refs:move_graph_nodes_to_cuda (Results 1 – 1 of 1) sorted by relevance

/aosp_15_r20/external/pytorch/torch/_dynamo/
H A Dcompiled_autograd.py255 def move_graph_nodes_to_cuda(self, graph) -> List[int]: member in AutogradCompilerInstance
313 runtime_inputs_to_move = self.move_graph_nodes_to_cuda(self.fx_tracer.graph)