Searched refs:temp_memory_allocated (Results 1 – 4 of 4) sorted by relevance
847 tracking_state_->temp_memory_allocated += out_temp->TotalBytes(); in allocate_temp()969 tracking_state_->temp_memory_allocated -= it->second; in set_output()1042 tracking_state_->temp_memory_allocated += size; in record_temp_memory_allocation()1048 int64 OpKernelContext::temp_memory_allocated() const { in temp_memory_allocated() function in tensorflow::OpKernelContext1051 return tracking_state_->temp_memory_allocated; in temp_memory_allocated()1090 tracking_state_->temp_memory_allocated = 0; in clear_recorded_memory()
1245 int64 temp_memory_allocated() const LOCKS_EXCLUDED(tracking_state_->stats_mu);1345 int64 temp_memory_allocated GUARDED_BY(stats_mu) = 0;
154 ms->set_temp_memory_size(ctx->temp_memory_allocated()); in SetMemory()
1725 ctx->persistent_memory_allocated() + ctx->temp_memory_allocated(); in MaybeQueue()