Home
last modified time | relevance | path

Searched refs:TorchFunctionModeStackVariable (Results 1 – 5 of 5) sorted by relevance

/aosp_15_r20/external/pytorch/torch/_dynamo/variables/
H A Dtorch.py51 TorchFunctionModeStackVariable,
787 TorchFunctionModeStackVariable.register_mutation(tx)
795 TorchFunctionModeStackVariable.register_mutation(tx)
817 TorchFunctionModeStackVariable.register_mutation(tx)
819 TorchFunctionModeStackVariable.clear_default_device(tx)
821 TorchFunctionModeStackVariable.register_device_context_insertion(tx)
H A Dtorch_function.py62 class TorchFunctionModeStackVariable(VariableTracker): class
/aosp_15_r20/external/pytorch/torch/_dynamo/
H A Dsource.py614 from .variables.torch_function import TorchFunctionModeStackVariable
616 return TorchFunctionModeStackVariable.get_mode_index(self.ind)
H A Dside_effects.py594 var, variables.torch_function.TorchFunctionModeStackVariable
H A Dsymbolic_convert.py2766 from .variables.torch_function import TorchFunctionModeStackVariable
2768 TorchFunctionModeStackVariable.reset()