Home
last modified time | relevance | path

Searched refs:UnspecializedNNModuleVariable (Results 1 – 9 of 9) sorted by relevance

/aosp_15_r20/external/pytorch/torch/_dynamo/variables/
H A Dnn_module.py299 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__.py92 UnspecializedNNModuleVariable,
H A Duser_defined.py1023 self, variables.UnspecializedNNModuleVariable
1039 variables.UnspecializedNNModuleVariable,
1145 and isinstance(self, variables.UnspecializedNNModuleVariable)
H A Ddicts.py53 variables.UnspecializedNNModuleVariable,
93 elif isinstance(self.vt, variables.UnspecializedNNModuleVariable):
H A Dbuilder.py191 UnspecializedNNModuleVariable,
1396 result = UnspecializedNNModuleVariable(value, source=self.source)
1406 return UnspecializedNNModuleVariable(value, source=self.get_source())
H A Dbuiltin.py1653 if isinstance(obj, variables.UnspecializedNNModuleVariable):
/aosp_15_r20/external/pytorch/torch/_dynamo/
H A Dside_effects.py286 variable_cls = variables.UnspecializedNNModuleVariable
H A Dsymbolic_convert.py121 from .variables.nn_module import NNModuleVariable, UnspecializedNNModuleVariable
489 elif isinstance(value, UnspecializedNNModuleVariable):
H A Doutput_graph.py838 return variables.UnspecializedNNModuleVariable(target, **options)