Home
last modified time | relevance | path

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

/aosp_15_r20/external/pytorch/torch/_dynamo/variables/
H A Dnn_module.py26 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 Dsource.py568 class NNModuleSource(ChainedSource): class
580 class UnspecializedNNModuleSource(NNModuleSource):
592 class FSDPNNModuleSource(NNModuleSource):
H A Dguards.py93 NNModuleSource,
895 NNModuleSource,
1181 if isinstance(source, NNModuleSource):