FSDPParamGroupUseTrainingStateVariable (reference) in projects: aosp_15_r20

Searched refs:FSDPParamGroupUseTrainingStateVariable (Results 1 – 3 of 3) sorted by relevance

/aosp_15_r20/external/pytorch/torch/_dynamo/variables/
H A D	__init__.py	`11 FSDPParamGroupUseTrainingStateVariable,`
H A D	ctx_manager.py	`919 class FSDPParamGroupUseTrainingStateVariable(ContextWrappingVariable): class 924 var = FSDPParamGroupUseTrainingStateVariable(`
H A D	torch.py	`221 FSDPParamGroupUseTrainingStateVariable, 324 return FSDPParamGroupUseTrainingStateVariable.create(`