Home
last modified time | relevance | path

Searched refs:patch_tracer (Results 1 – 2 of 2) sorted by relevance

/aosp_15_r20/external/pytorch/torch/distributed/fsdp/
H A D_trace_utils.py95 def patch_tracer(self, tracer: torch.fx.Tracer, root_module: nn.Module): member in _ExecOrderTracer
/aosp_15_r20/external/pytorch/test/distributed/fsdp/
H A Dtest_fsdp_fx.py51 with exec_order_tracer.patch_tracer(tracer=tracer, root_module=model):