Searched refs:_annotate_modules_for_dynamo (Results 1 – 3 of 3) sorted by relevance
11 from torch.distributed.fsdp._dynamo_utils import _annotate_modules_for_dynamo73 _annotate_modules_for_dynamo(module, state._ignored_modules, True)
7 def _annotate_modules_for_dynamo( function
41 from torch.distributed.fsdp._dynamo_utils import _annotate_modules_for_dynamo446 _annotate_modules_for_dynamo(module, self._ignored_modules, use_orig_params)