Searched defs:_TorchDispatchModeKey (Results 1 – 2 of 2) sorted by relevance
208 def _pop_mode(k: Optional[Union[DispatchKey, torch._C._TorchDispatchModeKey]] = None):
241 class _TorchDispatchModeKey(Enum): class