| /aosp_15_r20/external/pytorch/torch/_dynamo/ |
| H A D | device_interface.py | 59 def current_device() -> int: member in DeviceInterface.Worker
67 def current_device(): member in DeviceInterface
168 def current_device() -> int: member in CudaInterface.Worker
193 current_device = staticmethod(torch.cuda.current_device) variable in CudaInterface
239 def current_device() -> int: member in XpuInterface.Worker
264 current_device = staticmethod(torch.xpu.current_device) variable in XpuInterface
|
| /aosp_15_r20/external/pytorch/c10/cuda/ |
| H A D | CUDAGuard.h | 63 Device current_device() const { in current_device() function
126 std::optional<Device> current_device() const { in current_device() function
190 Device current_device() const { in current_device() function
|
| H A D | CUDAFunctions.cpp | 126 DeviceIndex current_device() { in current_device() function
|
| /aosp_15_r20/external/pytorch/c10/core/ |
| H A D | DeviceGuard.h | 77 Device current_device() const { in current_device() function
177 std::optional<Device> current_device() const { in current_device() function
|
| H A D | StreamGuard.h | 72 Device current_device() const { in current_device() function
|
| /aosp_15_r20/external/pytorch/c10/core/impl/ |
| H A D | InlineDeviceGuard.h | 191 Device current_device() const { in current_device() function
415 std::optional<Device> current_device() const { in current_device() function
|
| H A D | InlineStreamGuard.h | 109 Device current_device() const { in current_device() function
|
| /aosp_15_r20/external/pytorch/aten/src/ATen/hip/impl/ |
| H A D | HIPGuardImplMasqueradingAsCUDA.h | 265 Device current_device() const { return guard_.current_device(); } in current_device() function
285 std::optional<Device> current_device() const { return guard_.current_device(); } in current_device() function
309 Device current_device() const { return guard_.current_device(); } in current_device() function
|
| /aosp_15_r20/external/pytorch/test/inductor/extension_backends/triton/ |
| H A D | device_interface.py | 60 def current_device() -> int: member in DeviceInterface.Worker
71 def current_device() -> int: member in DeviceInterface
|
| /aosp_15_r20/external/pytorch/test/cpp_extensions/ |
| H A D | mtia_extension.cpp | 12 static thread_local c10::DeviceIndex current_device = 0; variable
41 c10::Device current_device = getDevice(); in setDevice() local
|
| /aosp_15_r20/external/pytorch/aten/src/ATen/cuda/detail/ |
| H A D | CUDAHooks.cpp | 235 DeviceIndex current_device() { in current_device() function
244 DeviceIndex CUDAHooks::current_device() const { in current_device() function in at::cuda::detail::CUDAHooks
|
| /aosp_15_r20/external/pytorch/torch/csrc/xpu/ |
| H A D | Module.cpp | 69 auto current_device = c10::xpu::exchange_device(device_index); in THXPModule_exchangeDevice_wrap() local
86 auto current_device = c10::xpu::maybe_exchange_device(device_index); in THXPModule_maybeExchangeDevice_wrap() local
|
| H A D | Stream.cpp | 19 const auto current_device = c10::xpu::current_device(); in THXPStream_pynew() local
|
| /aosp_15_r20/external/pytorch/aten/src/ATen/detail/ |
| H A D | HIPHooksInterface.h | 42 virtual c10::DeviceIndex current_device() const { in current_device() function
|
| H A D | XPUHooksInterface.h | 49 virtual DeviceIndex current_device() const { in current_device() function
|
| H A D | CUDAHooksInterface.h | 120 virtual DeviceIndex current_device() const { in current_device() function
|
| /aosp_15_r20/external/pytorch/aten/src/ATen/xpu/detail/ |
| H A D | XPUHooks.cpp | 60 DeviceIndex XPUHooks::current_device() const { in current_device() function in at::xpu::detail::XPUHooks
|
| /aosp_15_r20/external/pytorch/torch/csrc/cuda/ |
| H A D | Stream.cpp | 22 const auto current_device = c10::cuda::current_device(); in THCPStream_pynew() local
|
| /aosp_15_r20/frameworks/base/libs/usb/tests/accessorytest/ |
| H A D | usb.c | 30 static struct usb_device *current_device = NULL; variable
|
| /aosp_15_r20/external/pytorch/torch/cpu/ |
| H A D | __init__.py | 179 def current_device() -> str: function
|
| /aosp_15_r20/external/pytorch/aten/src/ATen/mps/ |
| H A D | MPSGuardImpl.h | 162 std::optional<Device> current_device() const { in current_device() function
|
| /aosp_15_r20/external/pytorch/c10/xpu/ |
| H A D | XPUFunctions.cpp | 185 DeviceIndex current_device() { in current_device() function
|
| /aosp_15_r20/external/pytorch/torch/mtia/ |
| H A D | __init__.py | 127 def current_device() -> int: function
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/mlir/lite/experimental/tac/transforms/ |
| H A D | get_alternative_subgraph.cc | 151 auto current_device = GetTargetAnnotation(func); in GetAlternativeGraphForFunc() local
|
| /aosp_15_r20/external/pytorch/test/ |
| H A D | test_cpp_extensions_open_device_registration.py | 58 def current_device(): function
|