Searched refs:run_and_get_graph_lowering (Results 1 – 2 of 2) sorted by relevance
| /aosp_15_r20/external/pytorch/test/inductor/ |
| H A D | test_cuda_repro.py | 20 run_and_get_graph_lowering,
1253 compiled_fn, (graph,) = run_and_get_graph_lowering(
1289 result, (graph,) = run_and_get_graph_lowering(
1296 result, (graph,) = run_and_get_graph_lowering(fn, inp)
1300 result, (graph,) = run_and_get_graph_lowering(lambda: result.sum().backward())
|
| /aosp_15_r20/external/pytorch/torch/_inductor/ |
| H A D | utils.py | 1368 def run_and_get_graph_lowering(fn, *args, **kwargs): function
|