Searched defs:PrivateUse1 (Results 1 – 7 of 7) sorted by relevance
| /aosp_15_r20/external/pytorch/c10/core/ |
| H A D | DeviceType.h | 57 PrivateUse1 = 20, // PrivateUse1 device enumerator
|
| H A D | Backend.h | 64 PrivateUse1, enumerator
|
| /aosp_15_r20/external/pytorch/test/cpp_extensions/ |
| H A D | open_registration_extension.cpp | 506 TORCH_LIBRARY_IMPL(aten, PrivateUse1, m) { in TORCH_LIBRARY_IMPL() argument
528 TORCH_LIBRARY_IMPL(aten, PrivateUse1, m) { in TORCH_LIBRARY_IMPL() argument
|
| /aosp_15_r20/external/pytorch/test/cpp_extensions/open_registration_extension/pytorch_openreg/csrc/ |
| H A D | OpenRegMem.cpp | 116 TORCH_LIBRARY_IMPL(aten, PrivateUse1, m) { in TORCH_LIBRARY_IMPL() argument
|
| /aosp_15_r20/external/pytorch/torch/csrc/profiler/orchestration/ |
| H A D | observer.h | 20 PrivateUse1, // PrivateUse1 kernels, runtime enumerator
|
| /aosp_15_r20/external/pytorch/test/inductor/extension_backends/cpp/ |
| H A D | extension_device.cpp | 154 TORCH_LIBRARY_IMPL(aten, PrivateUse1, m) { in TORCH_LIBRARY_IMPL() argument
|
| /aosp_15_r20/external/pytorch/torchgen/ |
| H A D | model.py | 147 PrivateUse1 = auto() variable in DispatchKey
|