Searched refs:tracking_state_ (Results 1 – 2 of 2) sorted by relevance
332 tracking_state_ = absl::make_unique<TrackingState>(); in OpKernelContext()354 !tracking_state_->wrapped_allocators.empty()) { in ~OpKernelContext()357 for (auto& wrapped_allocator : tracking_state_->wrapped_allocators) { in ~OpKernelContext()372 DCHECK(tracking_state_); in get_allocator()373 mutex_lock lock(tracking_state_->mu); in get_allocator()374 for (const auto& wrapped : tracking_state_->wrapped_allocators) { in get_allocator()381 tracking_state_->wrapped_allocators.push_back( in get_allocator()806 DCHECK(tracking_state_); in allocate_temp()807 mutex_lock l(tracking_state_->stats_mu); in allocate_temp()808 tracking_state_->temp_memory_allocated += out_temp->TotalBytes(); in allocate_temp()[all …]
1092 if (tracking_state_) { in ConsumeWrappedAllocators()1093 mutex_lock lock(tracking_state_->mu); in ConsumeWrappedAllocators()1094 retrieved.swap(tracking_state_->wrapped_allocators); in ConsumeWrappedAllocators()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);1326 std::unique_ptr<TrackingState> tracking_state_; variable