Searched refs:_get_current_function_mode_stack (Results 1 – 3 of 3) sorted by relevance
12 from torch.overrides import _get_current_function_mode_stack, BaseTorchFunctionMode181 stack = _get_current_function_mode_stack()194 stack = _get_current_function_mode_stack()211 stack = _get_current_function_mode_stack()226 stack = _get_current_function_mode_stack()237 stack = _get_current_function_mode_stack()
24 _get_current_function_mode_stack,1328 self.assertEqual(_get_current_function_mode_stack(), [])1331 self.assertEqual(_get_current_function_mode_stack(), [mode1])1335 self.assertEqual(_get_current_function_mode_stack(), [mode1, mode2])
2056 def _get_current_function_mode_stack(): function