Home
last modified time | relevance | path

Searched defs:BackendSelect (Results 1 – 4 of 4) sorted by relevance

/aosp_15_r20/external/pytorch/aten/src/ATen/core/
H A DBackendSelectFallbackKernel.cpp3 TORCH_LIBRARY_IMPL(_, BackendSelect, m) { in TORCH_LIBRARY_IMPL() argument
/aosp_15_r20/external/pytorch/c10/core/
H A DDispatchKey.h228 BackendSelect, enumerator
/aosp_15_r20/external/pytorch/test/cpp/jit/
H A Dtest_custom_class_registrations.cpp723 TORCH_LIBRARY_IMPL(_TorchScriptTesting, BackendSelect, m) { in TORCH_LIBRARY_IMPL() argument
/aosp_15_r20/external/pytorch/torchgen/
H A Dmodel.py106 BackendSelect = auto() variable in DispatchKey