Home
last modified time | relevance | path

Searched defs:DeviceCachingAllocator (Results 1 – 3 of 3) sorted by relevance

/aosp_15_r20/external/pytorch/aten/src/ATen/test/
H A Dcuda_reportMemoryUsage_test.cpp7 TEST(DeviceCachingAllocator, check_reporter) { in TEST() argument
/aosp_15_r20/external/pytorch/c10/xpu/
H A DXPUCachingAllocator.cpp127 class DeviceCachingAllocator { class
402 DeviceCachingAllocator(DeviceIndex device_index) in DeviceCachingAllocator() function in c10::xpu::XPUCachingAllocator::DeviceCachingAllocator
/aosp_15_r20/external/pytorch/c10/cuda/
H A DCUDACachingAllocator.cpp1014 class DeviceCachingAllocator { class
1089 DeviceCachingAllocator() in DeviceCachingAllocator() function in c10::cuda::CUDACachingAllocator::Native::DeviceCachingAllocator