Home
last modified time | relevance | path

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

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