Home
last modified time | relevance | path

Searched defs:exchangeDevice (Results 1 – 15 of 15) sorted by relevance

/aosp_15_r20/external/pytorch/test/cpp_extensions/
H A Dmtia_extension.cpp29 c10::Device exchangeDevice(c10::Device d) const override { in exchangeDevice() function
162 c10::DeviceIndex exchangeDevice(c10::DeviceIndex device) const override { in exchangeDevice() function
H A Dmaia_extension.cpp73 Device exchangeDevice(Device d) const override { in exchangeDevice() function
/aosp_15_r20/external/pytorch/aten/src/ATen/detail/
H A DAcceleratorHooksInterface.h35 virtual DeviceIndex exchangeDevice(DeviceIndex device) const { in exchangeDevice() function
H A DMTIAHooksInterface.h70 DeviceIndex exchangeDevice(DeviceIndex device) const override { in exchangeDevice() function
/aosp_15_r20/external/pytorch/aten/src/ATen/native/metal/
H A DMetalGuardImpl.cpp17 Device exchangeDevice(Device) const override { in exchangeDevice() function
/aosp_15_r20/external/pytorch/aten/src/ATen/native/vulkan/
H A DVulkanGuardImpl.cpp20 Device exchangeDevice(Device) const override { in exchangeDevice() function
/aosp_15_r20/external/pytorch/c10/core/impl/
H A DFakeGuardImpl.h27 Device exchangeDevice(Device d) const override { in exchangeDevice() function
H A DVirtualGuardImpl.h27 Device exchangeDevice(Device d) const override { in exchangeDevice() function
H A DDeviceGuardImplInterface.h249 Device exchangeDevice(Device) const override { in exchangeDevice() function
/aosp_15_r20/external/pytorch/aten/src/ATen/mps/
H A DMPSGuardImpl.h47 Device exchangeDevice(Device d) const override { in exchangeDevice() function
/aosp_15_r20/external/pytorch/c10/xpu/impl/
H A DXPUGuardImpl.h27 Device exchangeDevice(Device d) const override { in exchangeDevice() function
/aosp_15_r20/external/pytorch/torch/csrc/lazy/core/
H A Dtensor_impl.cpp28 c10::Device exchangeDevice(c10::Device device) const override { in exchangeDevice() function
/aosp_15_r20/external/pytorch/c10/cuda/impl/
H A DCUDAGuardImpl.h33 Device exchangeDevice(Device d) const override { in exchangeDevice() function
/aosp_15_r20/external/pytorch/test/cpp_extensions/open_registration_extension/pytorch_openreg/csrc/
H A DOpenRegHooks.cpp50 c10::Device exchangeDevice(c10::Device d) const override { in exchangeDevice() function
/aosp_15_r20/external/pytorch/aten/src/ATen/hip/impl/
H A DHIPGuardImplMasqueradingAsCUDA.h65 Device exchangeDevice(Device d) const override { in exchangeDevice() function