Searched refs:persistent_memory_allocated (Results 1 – 5 of 5) sorted by relevance
76 EXPECT_EQ(ctx.persistent_memory_allocated(), 512); in PersistentMemoryTrackingTest()78 EXPECT_EQ(ctx.persistent_memory_allocated(), 480); in PersistentMemoryTrackingTest()
201 ms->set_persistent_memory_size(context->persistent_memory_allocated()); in UpdateStats()
150 ms->set_persistent_memory_size(ctx->persistent_memory_allocated()); in SetMemory()
1201 int64 persistent_memory_allocated() const LOCKS_EXCLUDED(stats_mu_);
912 int64 OpKernelContext::persistent_memory_allocated() const { in persistent_memory_allocated() function in tensorflow::OpKernelContext