Searched refs:temp_memory_allocated (Results 1 – 4 of 4) sorted by relevance
196 ms->set_temp_memory_size(context->temp_memory_allocated()); in UpdateStats()
146 ms->set_temp_memory_size(ctx->temp_memory_allocated()); in SetMemory()
1193 int64 temp_memory_allocated() const LOCKS_EXCLUDED(stats_mu_);
895 int64 OpKernelContext::temp_memory_allocated() const { in temp_memory_allocated() function in tensorflow::OpKernelContext