Searched refs:get_nn_module_stack_source (Results 1 – 2 of 2) sorted by relevance
| /aosp_15_r20/external/pytorch/torch/_dynamo/variables/ |
| H A D | nn_module.py | 144 def get_nn_module_stack_source(self): member in NNModuleVariable
304 AttrSource(self.get_nn_module_stack_source(), name)
350 self.module_key, self.get_nn_module_stack_source(), tx, mod
496 self.module_key, self.get_nn_module_stack_source(), tx, module
809 def get_nn_module_stack_source(self): member in UnspecializedNNModuleVariable
893 str(id(mod)), self.get_nn_module_stack_source(), tx, mod
|
| H A D | user_defined.py | 1047 AttrSource(self.get_nn_module_stack_source(), name)
|