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):
|