Searched refs:on_compile_start (Results 1 – 3 of 3) sorted by relevance
79 @torch._dynamo.on_compile_start114 @torch._dynamo.on_compile_start
70 def on_compile_start(callback): function
4 from .callback import callback_handler, on_compile_end, on_compile_start