Searched refs:functionality_key (Results 1 – 3 of 3) sorted by relevance
582 auto functionality_key = variable584 if (isPerBackendFunctionalityKey(functionality_key)) {586 functionality_key,594 toString(functionality_key),604 return functionality_key;
370 … functionality_key = torch._C._to_functionality_key(dispatch_key) # type: ignore[attr-defined]371 if functionality_key == DispatchKey.PreDispatch:389 with _pop_mode_temporarily(functionality_key) as mode:811 functionality_key = torch._C._to_functionality_key(key) # type: ignore[attr-defined]812 if functionality_key == DispatchKey.PreDispatch:
196 auto functionality_key = toFunctionalityKey(tid); in TEST() local202 DispatchKeySet(functionality_key) | DispatchKeySet(backend_key); in TEST()