Home
last modified time | relevance | path

Searched defs:_guards_singleton (Results 1 – 1 of 1) sorted by relevance

/aosp_15_r20/external/pytorch/torch/_dynamo/variables/
H A Dctx_manager.py235 _guards_singleton = Guard(GlobalStateSource(), GuardBuilder.FUNCTORCH_STACK_MATCH) variable in JvpIncrementNestingCtxManagerVariable
309 _guards_singleton = Guard(GlobalStateSource(), GuardBuilder.DUAL_LEVEL) variable in DualLevelContextManager
352 _guards_singleton = Guard(GlobalStateSource(), GuardBuilder.FUNCTORCH_STACK_MATCH) variable in GradIncrementNestingCtxManagerVariable
422 _guards_singleton = Guard(GlobalStateSource(), GuardBuilder.FUNCTORCH_STACK_MATCH) variable in VmapIncrementNestingCtxManagerVariable
457 _guards_singleton = Guard(GlobalStateSource(), GuardBuilder.GRAD_MODE) variable in GradModeVariable
615 _guards_singleton = Guard(GlobalStateSource(), GuardBuilder.TORCH_FUNCTION_STATE) variable in TorchFunctionDisableVariable
646 _guards_singleton = Guard( variable in DeterministicAlgorithmsVariable
920 _guards_singleton = Guard(GlobalStateSource(), GuardBuilder.FSDP_TRAINING_STATE) variable in FSDPParamGroupUseTrainingStateVariable