Searched refs:aot_graph_input_parser (Results 1 – 5 of 5) sorted by relevance
8 from torch._dynamo.debug_utils import aot_graph_input_parser149 kwargs = aot_graph_input_parser(forward, device="cuda")165 kwargs = aot_graph_input_parser(
844 from torch._dynamo.debug_utils import aot_graph_input_parser848 kwargs = aot_graph_input_parser(forward)
9 from torch._dynamo.debug_utils import aot_graph_input_parser235 kwargs = aot_graph_input_parser(forward, device="cuda")
33 from torch._dynamo.debug_utils import aot_graph_input_parser8450 kwargs = aot_graph_input_parser(forward, device=GPU_TYPE)
690 def aot_graph_input_parser( function