Home
last modified time | relevance | path

Searched defs:getNumGPUs (Results 1 – 7 of 7) sorted by relevance

/aosp_15_r20/external/pytorch/aten/src/ATen/cuda/
H A DCUDAContextLight.h59 inline int64_t getNumGPUs() { in getNumGPUs() function
/aosp_15_r20/external/pytorch/aten/src/ATen/detail/
H A DHIPHooksInterface.h58 virtual int getNumGPUs() const { in getNumGPUs() function
H A DXPUHooksInterface.h45 virtual DeviceIndex getNumGPUs() const { in getNumGPUs() function
H A DCUDAHooksInterface.h185 virtual int getNumGPUs() const { in getNumGPUs() function
/aosp_15_r20/external/pytorch/aten/src/ATen/xpu/detail/
H A DXPUHooks.cpp56 c10::DeviceIndex XPUHooks::getNumGPUs() const { in getNumGPUs() function in at::xpu::detail::XPUHooks
/aosp_15_r20/external/pytorch/aten/src/ATen/cuda/detail/
H A DCUDAHooks.cpp439 int CUDAHooks::getNumGPUs() const { in getNumGPUs() function in at::cuda::detail::CUDAHooks
/aosp_15_r20/external/pytorch/aten/src/ATen/
H A DContext.h476 static inline size_t getNumGPUs() { in getNumGPUs() function