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()
1076 tracking_state_->persistent_memory_allocated += size; in record_persistent_memory_allocation()1083 int64_t OpKernelContext::persistent_memory_allocated() const { in persistent_memory_allocated() function in tensorflow::OpKernelContext1086 return tracking_state_->persistent_memory_allocated; in persistent_memory_allocated()1106 tracking_state_->persistent_memory_allocated = 0; in clear_recorded_memory()
1194 int64_t persistent_memory_allocated() const1282 int64_t persistent_memory_allocated TF_GUARDED_BY(stats_mu) = 0;
158 ms->set_persistent_memory_size(ctx->persistent_memory_allocated()); in SetMemory()
2070 ctx->persistent_memory_allocated() + ctx->temp_memory_allocated(); in MaybeQueue()