Home
last modified time | relevance | path

Searched refs:functionality_key (Results 1 – 3 of 3) sorted by relevance

/aosp_15_r20/external/pytorch/c10/core/
H A DDispatchKeySet.h582 auto functionality_key = variable
584 if (isPerBackendFunctionalityKey(functionality_key)) {
586 functionality_key,
594 toString(functionality_key),
604 return functionality_key;
/aosp_15_r20/external/pytorch/torch/
H A D_ops.py370functionality_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:
/aosp_15_r20/external/pytorch/c10/test/core/
H A DDispatchKeySet_test.cpp196 auto functionality_key = toFunctionalityKey(tid); in TEST() local
202 DispatchKeySet(functionality_key) | DispatchKeySet(backend_key); in TEST()