Home
last modified time | relevance | path

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

/aosp_15_r20/external/pytorch/torch/_dynamo/variables/
H A Dnn_module.py24 FSDPNNModuleSource,
1174 attr_source, (FSDPNNModuleSource, UnspecializedNNModuleSource)
1177 return FSDPNNModuleSource(attr_source)
/aosp_15_r20/external/pytorch/torch/_dynamo/
H A Dsource.py592 class FSDPNNModuleSource(NNModuleSource): class
H A Dguards.py86 FSDPNNModuleSource,
898 FSDPNNModuleSource,