Searched refs:count_mutex_ (Results 1 – 4 of 4) sorted by relevance
67 mutable tensorflow::mutex count_mutex_;70 int64 allocation_count_ TF_GUARDED_BY(count_mutex_) = 0;71 int64 deallocation_count_ TF_GUARDED_BY(count_mutex_) = 0;74 std::map<int, int64> device_allocation_count_ TF_GUARDED_BY(count_mutex_);75 std::map<int, int64> device_deallocation_count_ TF_GUARDED_BY(count_mutex_);
50 tensorflow::mutex_lock lock(count_mutex_); in Allocate()61 tensorflow::mutex_lock lock(count_mutex_); in Deallocate()69 tensorflow::mutex_lock lock(count_mutex_); in allocation_count()74 tensorflow::mutex_lock lock(count_mutex_); in allocation_count()84 tensorflow::mutex_lock lock(count_mutex_); in deallocation_count()89 tensorflow::mutex_lock lock(count_mutex_); in deallocation_count()
36 std::lock_guard<std::mutex> lock(count_mutex_); in DecrementCount()46 ruy::Wait(condition, spin_duration, &count_cond_, &count_mutex_); in Wait()
61 std::mutex count_mutex_; variable