Home
last modified time | relevance | path

Searched refs:release_lock_on_malloc (Results 1 – 4 of 4) sorted by relevance

/aosp_15_r20/external/pytorch/c10/cuda/
H A DCUDACachingAllocator.h159 bool release_lock_on_malloc; member
H A DCUDACachingAllocator.cpp3392 md.release_lock_on_malloc = in snapshot()
/aosp_15_r20/external/pytorch/torch/csrc/cuda/
H A Dmemory_snapshot.cpp386 snapshot.config_metadata.release_lock_on_malloc); in _memory_snapshot_pickled()
H A DModule.cpp833 snapshot.config_metadata.release_lock_on_malloc; in THCPModule_memorySnapshot()