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()
1065 tracking_state_->persistent_memory_allocated += size; in record_persistent_memory_allocation()1072 int64 OpKernelContext::persistent_memory_allocated() const { in persistent_memory_allocated() function in tensorflow::OpKernelContext1075 return tracking_state_->persistent_memory_allocated; in persistent_memory_allocated()1095 tracking_state_->persistent_memory_allocated = 0; in clear_recorded_memory()
1233 int64 persistent_memory_allocated() const1321 int64 persistent_memory_allocated TF_GUARDED_BY(stats_mu) = 0;
158 ms->set_persistent_memory_size(ctx->persistent_memory_allocated()); in SetMemory()
1937 ctx->persistent_memory_allocated() + ctx->temp_memory_allocated(); in MaybeQueue()