Searched refs:persistent_memory_allocated (Results 1 – 5 of 5) sorted by relevance
80 EXPECT_EQ(ctx.persistent_memory_allocated(), 512); in PersistentMemoryTrackingTest()82 EXPECT_EQ(ctx.persistent_memory_allocated(), 480); in PersistentMemoryTrackingTest()
1061 tracking_state_->persistent_memory_allocated += size; in record_persistent_memory_allocation()1068 int64 OpKernelContext::persistent_memory_allocated() const { in persistent_memory_allocated() function in tensorflow::OpKernelContext1071 return tracking_state_->persistent_memory_allocated; in persistent_memory_allocated()1091 tracking_state_->persistent_memory_allocated = 0; in clear_recorded_memory()
1253 int64 persistent_memory_allocated() const1347 int64 persistent_memory_allocated GUARDED_BY(stats_mu) = 0;
158 ms->set_persistent_memory_size(ctx->persistent_memory_allocated()); in SetMemory()
1725 ctx->persistent_memory_allocated() + ctx->temp_memory_allocated(); in MaybeQueue()