Home
last modified time | relevance | path

Searched refs:attr_source (Results 1 – 5 of 5) sorted by relevance

/aosp_15_r20/external/pytorch/torch/_dynamo/variables/
H A Dnn_module.py804 def _wrap_source(self, attr_source): argument
805 if not isinstance(attr_source, UnspecializedNNModuleSource):
806 return UnspecializedNNModuleSource(attr_source)
807 return attr_source
1145 def _wrap_source(self, attr_source): argument
1146 if not isinstance(attr_source, UnspecializedBuiltinNNModuleSource):
1147 return UnspecializedBuiltinNNModuleSource(attr_source)
1148 return attr_source
1172 def _wrap_source(self, attr_source): argument
1174 attr_source, (FSDPNNModuleSource, UnspecializedNNModuleSource)
[all …]
H A Dtensor.py283 attr_source = AttrSource(self.source, name)
284 install_guard(attr_source.make_guard(GuardBuilder.HASATTR))
285 return VariableBuilder(tx, attr_source)(real_value)
H A Ddicts.py959 attr_source = AttrSource(self.source, name)
960 return VariableBuilder(tx, attr_source)(attr_value)
H A Dmisc.py484 attr_source = AttrSource(self.source, name)
485 return VariableBuilder(tx, attr_source)(attr_value)
/aosp_15_r20/external/pytorch/torch/_dynamo/
H A Dguards.py1171 attr_source = AttrSource(guard.originating_source, attr_name)
1174 attr_source, guard_fn, stack=guard.stack, user_stack=guard.user_stack