Home
last modified time | relevance | path

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

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