Searched refs:guard_source (Results 1 – 5 of 5) sorted by relevance
75 if source.guard_source() == GuardSource.CONSTANT:105 def guard_source(self): member in LocalSource119 def guard_source(self): member in SyntheticLocalSource130 def guard_source(self): member in RandomValueSource149 def guard_source(self): member in GlobalSource168 def guard_source(self): member in GlobalWeakRefSource181 def guard_source(self): member in WeakRefCallSource182 return self.base.guard_source()205 def guard_source(self): member in AttrSource206 return self.base.guard_source()[all …]
2214 guard_source = source.guard_source()2215 if guard_source is GuardSource.CONSTANT:
2053 guard_source: "torch._guards.GuardSource",2066 if guard_source.is_nn_module() and config.force_nn_module_property_static_shapes:
229 return self.originating_source.guard_source()837 def guard_source(self) -> GuardSource: member in Source844 if self.guard_source() is GuardSource.CONSTANT:849 return self.guard_source().is_specialized_nn_module()853 return self.guard_source() != GuardSource.SYNTHETIC_LOCAL
418 or source.guard_source() == GuardSource.CONSTANT1201 and self.source.guard_source().is_unspecialized_nn_module()1421 not self.source.guard_source().is_local()1425 or self.source.guard_source().is_specialized_nn_module()1426 or self.source.guard_source().is_unspecialized_builtin_nn_module()1474 or (source and source.guard_source().is_unspecialized_nn_module())1485 source.guard_source().is_specialized_nn_module() or make_graph_attribute1486 ) and not source.guard_source().is_fsdp_module():1757 if self.source.guard_source().is_unspecialized_nn_module():