Searched refs:PreDispatchTorchFunctionMode (Results 1 – 1 of 1) sorted by relevance
1096 if isinstance(mode, PreDispatchTorchFunctionMode):1236 class PreDispatchTorchFunctionMode(TorchFunctionMode): class1813 nullcontext, PreDispatchTorchFunctionMode1835 prev_proxy_function_mode: Union[nullcontext, PreDispatchTorchFunctionMode], argument1916 self.proxy_function_mode = PreDispatchTorchFunctionMode(fx_tracer)