Searched refs:pass_execution_and_save (Results 1 – 3 of 3) sorted by relevance
27 from ..utils import is_cpu_device, pass_execution_and_save140 pass_execution_and_save(147 pass_execution_and_save(153 pass_execution_and_save(159 pass_execution_and_save(165 pass_execution_and_save(171 pass_execution_and_save(177 pass_execution_and_save(183 pass_execution_and_save(189 pass_execution_and_save([all …]
7 from torch._inductor.utils import pass_execution_and_save30 pass_execution_and_save(fx_pass, gm, sample_input, "Apply testing pass")
1600 def pass_execution_and_save(func, gm, inp, msg): function