Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/core/kernels/
Dconstant_op_test.cc82 for (auto allocator_pair : ctx.ConsumeWrappedAllocators()) { in PersistentMemoryTrackingTest()
/external/tensorflow/tensorflow/core/common_runtime/eager/
Dkernel_and_device.cc180 for (const auto& allocator_pair : context->ConsumeWrappedAllocators()) { in UpdateStats()
/external/tensorflow/tensorflow/core/common_runtime/
Dstep_stats_collector.cc142 for (const auto& allocator_pair : ctx->ConsumeWrappedAllocators()) { in SetMemory()
/external/tensorflow/tensorflow/core/framework/
Dop_kernel.h1055 gtl::InlinedVector<WrappedAllocator, 4> ConsumeWrappedAllocators() { in ConsumeWrappedAllocators() function