Searched refs:UnspecializedNNModuleVariable (Results 1 – 9 of 9) sorted by relevance
/aosp_15_r20/external/pytorch/torch/_dynamo/variables/ |
H A D | nn_module.py | 299 if isinstance(out, (NNModuleVariable, UnspecializedNNModuleVariable)): 509 if isinstance(mod_var, UnspecializedNNModuleVariable): 763 class UnspecializedNNModuleVariable(UserDefinedObjectVariable): class 931 assert isinstance(module_var, UnspecializedNNModuleVariable) 936 assert isinstance(submodule_var, UnspecializedNNModuleVariable) 1032 variables.UnspecializedNNModuleVariable, 1140 class UnspecializedBuiltinNNModuleVariable(UnspecializedNNModuleVariable): 1151 class FSDPManagedNNModuleVariable(UnspecializedNNModuleVariable):
|
H A D | __init__.py | 92 UnspecializedNNModuleVariable,
|
H A D | user_defined.py | 1023 self, variables.UnspecializedNNModuleVariable 1039 variables.UnspecializedNNModuleVariable, 1145 and isinstance(self, variables.UnspecializedNNModuleVariable)
|
H A D | dicts.py | 53 variables.UnspecializedNNModuleVariable, 93 elif isinstance(self.vt, variables.UnspecializedNNModuleVariable):
|
H A D | builder.py | 191 UnspecializedNNModuleVariable, 1396 result = UnspecializedNNModuleVariable(value, source=self.source) 1406 return UnspecializedNNModuleVariable(value, source=self.get_source())
|
H A D | builtin.py | 1653 if isinstance(obj, variables.UnspecializedNNModuleVariable):
|
/aosp_15_r20/external/pytorch/torch/_dynamo/ |
H A D | side_effects.py | 286 variable_cls = variables.UnspecializedNNModuleVariable
|
H A D | symbolic_convert.py | 121 from .variables.nn_module import NNModuleVariable, UnspecializedNNModuleVariable 489 elif isinstance(value, UnspecializedNNModuleVariable):
|
H A D | output_graph.py | 838 return variables.UnspecializedNNModuleVariable(target, **options)
|