Home
last modified time | relevance | path

Searched defs:private_use_ks (Results 1 – 2 of 2) sorted by relevance

/aosp_15_r20/external/pytorch/test/inductor/extension_backends/cpp/
H A Dextension_device.cpp129 constexpr c10::DispatchKeySet private_use_ks(c10::DispatchKey::PrivateUse1); in custom_empty_memory_format() local
140 constexpr c10::DispatchKeySet private_use_ks(c10::DispatchKey::PrivateUse1); in custom_empty_strided() local
/aosp_15_r20/external/pytorch/test/cpp_extensions/
H A Dopen_registration_extension.cpp293 constexpr c10::DispatchKeySet private_use_ks(c10::DispatchKey::PrivateUse1); in custom_empty_memory_format() local
306 constexpr c10::DispatchKeySet private_use_ks(c10::DispatchKey::PrivateUse1); in custom_empty_symint() local
390 constexpr c10::DispatchKeySet private_use_ks(c10::DispatchKey::PrivateUse1); in custom_empty_strided() local