Searched refs:persistent_alloc_ids (Results 1 – 4 of 4) sorted by relevance
197 for (const auto& alloc_id : context->persistent_alloc_ids()) { in UpdateStats()
147 for (const auto& alloc_id : ctx->persistent_alloc_ids()) { in SetMemory()
1203 std::vector<int64> persistent_alloc_ids() const LOCKS_EXCLUDED(stats_mu_);
917 std::vector<int64> OpKernelContext::persistent_alloc_ids() const { in persistent_alloc_ids() function in tensorflow::OpKernelContext