Searched refs:persistent_alloc_ids (Results 1 – 3 of 3) sorted by relevance
1067 tracking_state_->persistent_alloc_ids.push_back(alloc_id); in record_persistent_memory_allocation()1081 std::vector<int64> OpKernelContext::persistent_alloc_ids() const { in persistent_alloc_ids() function in tensorflow::OpKernelContext1084 return std::vector<int64>(tracking_state_->persistent_alloc_ids.begin(), in persistent_alloc_ids()1085 tracking_state_->persistent_alloc_ids.end()); in persistent_alloc_ids()1097 tracking_state_->persistent_alloc_ids.clear(); in clear_recorded_memory()
1236 std::vector<int64> persistent_alloc_ids() const1324 gtl::InlinedVector<int64, 2> persistent_alloc_ids TF_GUARDED_BY(stats_mu);
155 for (const auto& alloc_id : ctx->persistent_alloc_ids()) { in SetMemory()