Home
last modified time | relevance | path

Searched refs:getAutocastRelatedKeySetFromBackend (Results 1 – 2 of 2) sorted by relevance

/aosp_15_r20/external/pytorch/c10/core/
H A DTensorImpl.cpp161 key_set = key_set | getAutocastRelatedKeySetFromBackend(k); in TensorImpl()
197 key_set_ - c10::getAutocastRelatedKeySetFromBackend(old_backend); in _change_backend_component_keys()
198 key_set = key_set | c10::getAutocastRelatedKeySetFromBackend(new_backend); in _change_backend_component_keys()
H A DDispatchKeySet.h859 inline DispatchKeySet getAutocastRelatedKeySetFromBackend(BackendComponent t) { in getAutocastRelatedKeySetFromBackend() function