Searched refs:temp_memory_allocated (Results 1 – 4 of 4) sorted by relevance
808 tracking_state_->temp_memory_allocated += out_temp->TotalBytes(); in allocate_temp()960 tracking_state_->temp_memory_allocated -= it->second; in maybe_track_allocations_for_set_output()1046 tracking_state_->temp_memory_allocated += size; in record_temp_memory_allocation()1052 int64 OpKernelContext::temp_memory_allocated() const { in temp_memory_allocated() function in tensorflow::OpKernelContext1055 return tracking_state_->temp_memory_allocated; in temp_memory_allocated()1094 tracking_state_->temp_memory_allocated = 0; in clear_recorded_memory()
1224 int64 temp_memory_allocated() const1319 int64 temp_memory_allocated TF_GUARDED_BY(stats_mu) = 0;
154 ms->set_temp_memory_size(ctx->temp_memory_allocated()); in SetMemory()
1937 ctx->persistent_memory_allocated() + ctx->temp_memory_allocated(); in MaybeQueue()