Searched defs:setCheckpointPoolState (Results 1 – 5 of 5) sorted by relevance
| /aosp_15_r20/external/pytorch/torch/csrc/cuda/ | ||
| H A D | CUDAPluggableAllocator.cpp | 322 setCheckpointPoolState( in setCheckpointPoolState() function in torch::cuda::CUDAPluggableAllocator::CUDAPluggableAllocator |
| /aosp_15_r20/external/pytorch/c10/cuda/ | ||
| H A D | CUDAMallocAsyncAllocator.cpp | 658 CheckpointDelta setCheckpointPoolState( in setCheckpointPoolState() function |
| H A D | CUDACachingAllocator.h | 365 inline CheckpointDelta setCheckpointPoolState( in setCheckpointPoolState() function |
| H A D | CUDACachingAllocator.cpp | 1841 RestoreResult setCheckpointPoolState(PrivatePoolState& pps) { in setCheckpointPoolState() function in c10::cuda::CUDACachingAllocator::Native::DeviceCachingAllocator |
| /aosp_15_r20/external/pytorch/test/ | ||
| H A D | test_cuda.py | 3999 def setCheckpointPoolState( member in TestBlockStateAbsorption |