Searched refs:persistent_alloc_ids_ (Results 1 – 4 of 4) sorted by relevance
905 if (!persistent_alloc_ids_) { in record_persistent_memory_allocation()906 persistent_alloc_ids_.reset(new gtl::InlinedVector<int64, 2>()); in record_persistent_memory_allocation()908 persistent_alloc_ids_->push_back(alloc_id); in record_persistent_memory_allocation()919 if (persistent_alloc_ids_) { in persistent_alloc_ids()920 return std::vector<int64>(persistent_alloc_ids_->begin(), in persistent_alloc_ids()921 persistent_alloc_ids_->end()); in persistent_alloc_ids()934 if (persistent_alloc_ids_) { in clear_recorded_memory()935 persistent_alloc_ids_->clear(); in clear_recorded_memory()
1275 std::unique_ptr<gtl::InlinedVector<int64, 2>> persistent_alloc_ids_
222 std::set<int64> persistent_alloc_ids_; variable
329 persistent_alloc_ids_.insert(alloc_id); in RecordMemoryStats()367 if (persistent_alloc_ids_.count(alloc_id) > 0) { in IsPersistentTensor()