Home
last modified time | relevance | path

Searched defs:uncheckedSetDevice (Results 1 – 13 of 13) sorted by relevance

/aosp_15_r20/external/pytorch/aten/src/ATen/native/metal/
H A DMetalGuardImpl.cpp27 void uncheckedSetDevice(Device d) const noexcept override { in uncheckedSetDevice() function
/aosp_15_r20/external/pytorch/aten/src/ATen/native/vulkan/
H A DVulkanGuardImpl.cpp30 void uncheckedSetDevice(Device d) const noexcept override { in uncheckedSetDevice() function
/aosp_15_r20/external/pytorch/c10/core/impl/
H A DFakeGuardImpl.h45 void uncheckedSetDevice(Device d) const noexcept override { in uncheckedSetDevice() function
H A DVirtualGuardImpl.h36 void uncheckedSetDevice(Device d) const noexcept override { in uncheckedSetDevice() function
H A DDeviceGuardImplInterface.h258 void uncheckedSetDevice(Device) const noexcept override { in uncheckedSetDevice() function
/aosp_15_r20/external/pytorch/aten/src/ATen/mps/
H A DMPSGuardImpl.h63 void uncheckedSetDevice(Device d) const noexcept override { in uncheckedSetDevice() function
/aosp_15_r20/external/pytorch/c10/xpu/impl/
H A DXPUGuardImpl.h43 void uncheckedSetDevice(Device d) const noexcept override { in uncheckedSetDevice() function
/aosp_15_r20/external/pytorch/torch/csrc/lazy/core/
H A Dtensor_impl.cpp44 void uncheckedSetDevice(c10::Device device) const noexcept override { in uncheckedSetDevice() function
/aosp_15_r20/external/pytorch/c10/cuda/impl/
H A DCUDAGuardImpl.h56 void uncheckedSetDevice(Device d) const noexcept override { in uncheckedSetDevice() function
/aosp_15_r20/external/pytorch/test/cpp_extensions/
H A Dmaia_extension.cpp85 void uncheckedSetDevice(Device d) const noexcept override { in uncheckedSetDevice() function
H A Dmtia_extension.cpp46 void uncheckedSetDevice(c10::Device d) const noexcept override { in uncheckedSetDevice() function
/aosp_15_r20/external/pytorch/test/cpp_extensions/open_registration_extension/pytorch_openreg/csrc/
H A DOpenRegHooks.cpp79 void uncheckedSetDevice(c10::Device d) const noexcept override { in uncheckedSetDevice() function
/aosp_15_r20/external/pytorch/aten/src/ATen/hip/impl/
H A DHIPGuardImplMasqueradingAsCUDA.h82 void uncheckedSetDevice(Device d) const noexcept override { in uncheckedSetDevice() function