Searched refs:NNModuleSource (Results 1 – 3 of 3) sorted by relevance
/aosp_15_r20/external/pytorch/torch/_dynamo/variables/ |
H A D | nn_module.py | 26 NNModuleSource, 169 source=NNModuleSource(GetItemSource(self.source, name)), 185 source=NNModuleSource(GetItemSource(self.source, idx)), 194 NNModuleSource(AttrSource(self.source, name)).make_guard( 297 out = VariableBuilder(tx, NNModuleSource(source))(subobj) 333 return VariableBuilder(tx, NNModuleSource(source))(subobj) 376 source=NNModuleSource(AttrSource(self.source, child_name)), 517 source=NNModuleSource(GetItemSource(self.source, key)), 550 source=NNModuleSource(gen_source(self.source, name)), 563 source=NNModuleSource(gen_source(self.source, name)), [all …]
|
/aosp_15_r20/external/pytorch/torch/_dynamo/ |
H A D | source.py | 568 class NNModuleSource(ChainedSource): class 580 class UnspecializedNNModuleSource(NNModuleSource): 592 class FSDPNNModuleSource(NNModuleSource):
|
H A D | guards.py | 93 NNModuleSource, 895 NNModuleSource, 1181 if isinstance(source, NNModuleSource):
|