Searched refs:stats_mu (Results 1 – 2 of 2) sorted by relevance
1221 TF_LOCKS_EXCLUDED(tracking_state_->stats_mu);1225 TF_LOCKS_EXCLUDED(tracking_state_->stats_mu);1230 TF_LOCKS_EXCLUDED(tracking_state_->stats_mu);1234 TF_LOCKS_EXCLUDED(tracking_state_->stats_mu);1237 TF_LOCKS_EXCLUDED(tracking_state_->stats_mu);1240 void clear_recorded_memory() TF_LOCKS_EXCLUDED(tracking_state_->stats_mu);1318 mutable mutex stats_mu; member1319 int64 temp_memory_allocated TF_GUARDED_BY(stats_mu) = 0;1321 int64 persistent_memory_allocated TF_GUARDED_BY(stats_mu) = 0;1323 temp_tensor_buffer_and_size TF_GUARDED_BY(stats_mu);[all …]
807 mutex_lock l(tracking_state_->stats_mu); in allocate_temp()951 mutex_lock l(tracking_state_->stats_mu); in maybe_track_allocations_for_set_output()1045 mutex_lock l(tracking_state_->stats_mu); in record_temp_memory_allocation()1054 mutex_lock l(tracking_state_->stats_mu); in temp_memory_allocated()1064 mutex_lock l(tracking_state_->stats_mu); in record_persistent_memory_allocation()1074 mutex_lock l(tracking_state_->stats_mu); in persistent_memory_allocated()1083 mutex_lock l(tracking_state_->stats_mu); in persistent_alloc_ids()1093 mutex_lock l(tracking_state_->stats_mu); in clear_recorded_memory()