Searched refs:enable_single_level_autograd_function (Results 1 – 3 of 3) sorted by relevance
23 def enable_single_level_autograd_function(): function
13 from torch._functorch.utils import enable_single_level_autograd_function92 with enable_single_level_autograd_function():
49 from torch._functorch.utils import enable_single_level_autograd_function5038 with enable_single_level_autograd_function():