Home
last modified time | relevance | path

Searched defs:getDeviceFromPtr (Results 1 – 9 of 9) sorted by relevance

/aosp_15_r20/external/pytorch/aten/src/ATen/cuda/
H A DCUDADevice.h10 inline Device getDeviceFromPtr(void* ptr) { in getDeviceFromPtr() function
/aosp_15_r20/external/pytorch/aten/src/ATen/xpu/
H A DXPUDevice.h8 inline Device getDeviceFromPtr(void* ptr) { in getDeviceFromPtr() function
/aosp_15_r20/external/pytorch/aten/src/ATen/detail/
H A DPrivateUse1HooksInterface.h21 virtual at::Device getDeviceFromPtr(void* data) const { in getDeviceFromPtr() function
H A DXPUHooksInterface.h53 virtual Device getDeviceFromPtr(void* /*data*/) const { in getDeviceFromPtr() function
H A DCUDAHooksInterface.h76 virtual Device getDeviceFromPtr(void* /*data*/) const { in getDeviceFromPtr() function
/aosp_15_r20/external/pytorch/aten/src/ATen/test/
H A Dxpu_device_test.cpp21 TEST(XpuDeviceTest, getDeviceFromPtr) { in TEST() argument
/aosp_15_r20/external/pytorch/aten/src/ATen/xpu/detail/
H A DXPUHooks.cpp46 Device XPUHooks::getDeviceFromPtr(void* data) const { in getDeviceFromPtr() function in at::xpu::detail::XPUHooks
/aosp_15_r20/external/pytorch/aten/src/ATen/cuda/detail/
H A DCUDAHooks.cpp120 Device CUDAHooks::getDeviceFromPtr(void* data) const { in getDeviceFromPtr() function in at::cuda::detail::CUDAHooks
/aosp_15_r20/external/pytorch/aten/src/ATen/
H A DContext.h79 Device getDeviceFromPtr(void* data, c10::DeviceType device_type) { in getDeviceFromPtr() function