Home
last modified time | relevance | path

Searched refs:allocator_mutex_ (Results 1 – 3 of 3) sorted by relevance

/aosp_15_r20/external/tensorflow/tensorflow/compiler/jit/
H A Dxla_device.cc97 mutex allocator_mutex_; // Guards the singleton allocator state. member in tensorflow::XlaDeviceAllocatorState
101 allocators_ TF_GUARDED_BY(allocator_mutex_);
117 mutex_lock lock(state.allocator_mutex_); in GetOrCreateXlaDeviceAllocator()
/aosp_15_r20/external/pytorch/torch/csrc/cuda/
H A DCUDAPluggableAllocator.cpp108 const std::lock_guard<std::mutex> lock(allocator_mutex_); in malloc()
150 const std::lock_guard<std::mutex> lock(allocator_mutex_); in raw_delete()
H A DCUDAPluggableAllocator.h177 std::mutex allocator_mutex_; member