Home
last modified time | relevance | path

Searched refs:NNModuleVariable (Results 1 – 10 of 10) sorted by relevance

/aosp_15_r20/external/pytorch/torch/_dynamo/variables/
H A Dbuiltin.py272 from .nn_module import NNModuleVariable
491 NNModuleVariable,
544 NNModuleVariable,
545 NNModuleVariable,
1294 from .nn_module import NNModuleVariable
1301 NNModuleVariable,
1713 if isinstance(obj, variables.NNModuleVariable):
1845 elif isinstance(obj, variables.NNModuleVariable):
1961 if len(args) > 0 and isinstance(args[0], variables.NNModuleVariable):
H A D__init__.py90 NNModuleVariable,
H A Dnn_module.py125 class NNModuleVariable(VariableTracker): class
299 if isinstance(out, (NNModuleVariable, UnspecializedNNModuleVariable)):
1031 variables.NNModuleVariable,
H A Dmisc.py652 if isinstance(node, variables.NNModuleVariable):
1004 if not isinstance(self.obj, variables.NNModuleVariable):
1042 variables.NNModuleVariable,
1070 variables.NNModuleVariable,
H A Ddicts.py52 variables.NNModuleVariable,
91 elif isinstance(self.vt, variables.NNModuleVariable):
H A Dfunctions.py328 self.obj, variables.NNModuleVariable
H A Duser_defined.py1040 variables.NNModuleVariable,
/aosp_15_r20/external/pytorch/torch/_dynamo/
H A Dcodegen.py27 from .variables.nn_module import NNModuleVariable
218 elif isinstance(value, NNModuleVariable):
H A Dsymbolic_convert.py121 from .variables.nn_module import NNModuleVariable, UnspecializedNNModuleVariable
482 elif isinstance(value, NNModuleVariable):
1752 if isinstance(obj, NNModuleVariable) and not isinstance(val, ConstantVariable):
3103 if args and isinstance(args[0], NNModuleVariable):
H A Doutput_graph.py100 from .variables.nn_module import NNModuleVariable
830 return NNModuleVariable(type(target), module_key, target, **options)