Home
last modified time | relevance | path

Searched defs:getPinnedMemoryAllocator (Results 1 – 12 of 12) sorted by relevance

/aosp_15_r20/external/pytorch/aten/src/ATen/xpu/
H A DPinnedMemoryAllocator.h8 inline TORCH_XPU_API at::Allocator* getPinnedMemoryAllocator() { in getPinnedMemoryAllocator() function
/aosp_15_r20/external/pytorch/aten/src/ATen/cuda/
H A DPinnedMemoryAllocator.h8 inline TORCH_CUDA_CPP_API at::Allocator* getPinnedMemoryAllocator() { in getPinnedMemoryAllocator() function
/aosp_15_r20/external/pytorch/aten/src/ATen/detail/
H A DAcceleratorHooksInterface.h49 virtual Allocator* getPinnedMemoryAllocator() const { in getPinnedMemoryAllocator() function
H A DPrivateUse1HooksInterface.h31 virtual Allocator* getPinnedMemoryAllocator() const override { in getPinnedMemoryAllocator() function
H A DHIPHooksInterface.h50 Allocator* getPinnedMemoryAllocator() const override { in getPinnedMemoryAllocator() function
H A DXPUHooksInterface.h61 Allocator* getPinnedMemoryAllocator() const override { in getPinnedMemoryAllocator() function
H A DMTIAHooksInterface.h98 Allocator* getPinnedMemoryAllocator() const override { in getPinnedMemoryAllocator() function
H A DMPSHooksInterface.h100 Allocator* getPinnedMemoryAllocator() const override { in getPinnedMemoryAllocator() function
H A DCUDAHooksInterface.h124 Allocator* getPinnedMemoryAllocator() const override { in getPinnedMemoryAllocator() function
/aosp_15_r20/external/pytorch/aten/src/ATen/xpu/detail/
H A DXPUHooks.cpp70 Allocator* XPUHooks::getPinnedMemoryAllocator() const { in getPinnedMemoryAllocator() function in at::xpu::detail::XPUHooks
/aosp_15_r20/external/pytorch/aten/src/ATen/cuda/detail/
H A DCUDAHooks.cpp252 Allocator* CUDAHooks::getPinnedMemoryAllocator() const { in getPinnedMemoryAllocator() function in at::cuda::detail::CUDAHooks
/aosp_15_r20/external/pytorch/test/cpp_extensions/
H A Dopen_registration_extension.cpp599 c10::Allocator* getPinnedMemoryAllocator() const override { in getPinnedMemoryAllocator() function