Searched refs:persistent_alloc_ids (Results 1 – 3 of 3) sorted by relevance
1078 tracking_state_->persistent_alloc_ids.push_back(alloc_id); in record_persistent_memory_allocation()1092 std::vector<int64_t> OpKernelContext::persistent_alloc_ids() const { in persistent_alloc_ids() function in tensorflow::OpKernelContext1095 return std::vector<int64_t>(tracking_state_->persistent_alloc_ids.begin(), in persistent_alloc_ids()1096 tracking_state_->persistent_alloc_ids.end()); in persistent_alloc_ids()1108 tracking_state_->persistent_alloc_ids.clear(); in clear_recorded_memory()
1197 std::vector<int64_t> persistent_alloc_ids() const1285 gtl::InlinedVector<int64_t, 2> persistent_alloc_ids TF_GUARDED_BY(stats_mu);
155 for (const auto& alloc_id : ctx->persistent_alloc_ids()) { in SetMemory()