Searched refs:PythonKeyTracer (Results 1 – 5 of 5) sorted by relevance
1005 class PythonKeyTracer(Tracer): class1501 class _ModuleStackTracer(PythonKeyTracer):1815 self.fx_tracer: Optional[PythonKeyTracer] = None1836 prev_fx_tracer: Optional[PythonKeyTracer], argument1860 self.fx_tracer = PythonKeyTracer()1936 def _create_sub_fx_tracer(parent_tracer: _ProxyTracer) -> PythonKeyTracer:1937 if type(parent_tracer) == PythonKeyTracer:1938 return PythonKeyTracer()
25 PythonKeyTracer,72 self.fx_tracer = PythonKeyTracer()95 self.fx_tracer.graph = torch.fx.Graph(tracer_cls=PythonKeyTracer)
11 PythonKeyTracer,
17 from torch.fx.experimental.proxy_tensor import PythonKeyTracer60 class ExportTracer(PythonKeyTracer):
40 from torch.fx.experimental.proxy_tensor import PythonKeyTracer169 class ExportTracer(PythonKeyTracer):