Home
last modified time | relevance | path

Searched refs:temp_memory_allocated (Results 1 – 4 of 4) sorted by relevance

/external/tensorflow/tensorflow/core/common_runtime/eager/
Dkernel_and_device.cc196 ms->set_temp_memory_size(context->temp_memory_allocated()); in UpdateStats()
/external/tensorflow/tensorflow/core/common_runtime/
Dstep_stats_collector.cc146 ms->set_temp_memory_size(ctx->temp_memory_allocated()); in SetMemory()
/external/tensorflow/tensorflow/core/framework/
Dop_kernel.h1193 int64 temp_memory_allocated() const LOCKS_EXCLUDED(stats_mu_);
Dop_kernel.cc895 int64 OpKernelContext::temp_memory_allocated() const { in temp_memory_allocated() function in tensorflow::OpKernelContext