Home
last modified time | relevance | path

Searched refs:TorchCtxManagerClassVariable (Results 1 – 4 of 4) sorted by relevance

/aosp_15_r20/external/pytorch/torch/_dynamo/variables/
H A D__init__.py104 from .torch import TorchCtxManagerClassVariable, TorchInGraphFunctionVariable
H A Duser_defined.py861 and variables.TorchCtxManagerClassVariable.is_matching_cls(
866 return variables.TorchCtxManagerClassVariable(
876 return variables.TorchCtxManagerClassVariable(
H A Dbuilder.py203 from .torch import TorchCtxManagerClassVariable, TorchInGraphFunctionVariable
980 elif TorchCtxManagerClassVariable.is_matching_cls(value):
982 return TorchCtxManagerClassVariable(value, source=self.source)
H A Dtorch.py190 class TorchCtxManagerClassVariable(BaseTorchVariable): class