Home
last modified time | relevance | path

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

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