Home
last modified time | relevance | path

Searched refs:bytes_in_use (Results 1 – 19 of 19) sorted by relevance

/external/tensorflow/tensorflow/core/framework/
Dallocator.cc36 this->bytes_limit ? *this->bytes_limit : 0, this->bytes_in_use, in DebugString()
126 stats_.bytes_in_use += alloc_size; in AllocateRaw()
128 std::max<int64>(stats_.peak_bytes_in_use, stats_.bytes_in_use); in AllocateRaw()
132 if (stats_.bytes_in_use > TotalAllocationWarningBytes() && in AllocateRaw()
135 LOG(WARNING) << "Total allocated memory " << stats_.bytes_in_use in AllocateRaw()
148 stats_.bytes_in_use -= alloc_size; in DeallocateRaw()
161 stats_.peak_bytes_in_use = stats_.bytes_in_use; in ClearStats()
Dallocator_test.cc27 static void CheckStats(Allocator* a, int64 num_allocs, int64 bytes_in_use, in CheckStats() argument
39 EXPECT_GT(stats->bytes_in_use, bytes_in_use - kSlop); in CheckStats()
40 EXPECT_LT(stats->bytes_in_use, bytes_in_use + kSlop); in CheckStats()
Dallocator.h62 int64 bytes_in_use; // Number of bytes in use. member
72 bytes_in_use(0), in AllocatorStats()
/external/tensorflow/tensorflow/core/common_runtime/
Dmkl_cpu_allocator.h95 stats_.bytes_in_use = 0; in ClearStats()
104 stats_.bytes_in_use += alloc_size; in IncrementStats()
106 std::max(stats_.peak_bytes_in_use, stats_.bytes_in_use); in IncrementStats()
114 stats_.bytes_in_use -= dealloc_size; in DecrementStats()
257 stats_.bytes_in_use = l_stats->bytes_in_use + s_stats->bytes_in_use; in GetStats()
Dbfc_allocator.cc337 stats_.bytes_in_use += chunk->size; in FindChunkPtr()
339 std::max(stats_.peak_bytes_in_use, stats_.bytes_in_use); in FindChunkPtr()
500 stats_.bytes_in_use -= c->size; in FreeAndMaybeCoalesce()
697 stats_.peak_bytes_in_use = stats_.bytes_in_use; in ClearStats()
Dstep_stats_collector.cc181 memory->set_allocator_bytes_in_use(stats->bytes_in_use); in AddAllocation()
/external/tensorflow/tensorflow/core/common_runtime/sycl/
Dsycl_allocator.cc50 stats_.bytes_in_use += bytes_allocated; in AllocateRaw()
52 std::max<int64>(stats_.max_bytes_in_use, stats_.bytes_in_use); in AllocateRaw()
64 stats_.bytes_in_use -= dealloc_size; in DeallocateRaw()
77 stats_.max_bytes_in_use = stats_.bytes_in_use; in ClearStats()
/external/tensorflow/tensorflow/contrib/memory_stats/python/kernel_tests/
Dmemory_stats_ops_test.py82 _, bytes_in_use, max_bytes_in_use = sess.run([c, bytes_in_use_op,
86 self.assertGreaterEqual(bytes_in_use, matrix_size_in_bytes * 1)
87 self.assertLess(bytes_in_use, matrix_size_in_bytes * 2)
/external/tensorflow/tensorflow/stream_executor/
Dallocator_stats.h31 int64 bytes_in_use; // Number of bytes in use. member
41 bytes_in_use(0), in AllocatorStats()
Dallocator_stats.cc28 this->bytes_limit ? *this->bytes_limit : 0, this->bytes_in_use, in DebugString()
/external/tensorflow/tensorflow/core/profiler/internal/
Dtfprof_timeline.cc168 for (const auto& bytes_in_use : node->node->allocator_bytes_in_use(step)) { in TrackNode() local
169 if (bytes_in_use.first <= 0) continue; in TrackNode()
170 dev.allocations[bytes_in_use.first] = bytes_in_use.second; in TrackNode()
Dtfprof_node.h180 std::map<int64, int64> bytes_in_use; in allocator_bytes_in_use() local
182 bytes_in_use[exec.memory_micros()] = exec.allocator_bytes_in_use(); in allocator_bytes_in_use()
184 return bytes_in_use; in allocator_bytes_in_use()
/external/tensorflow/tensorflow/contrib/memory_stats/python/ops/
Dmemory_stats_ops.py31 return gen_memory_stats_ops.bytes_in_use()
/external/tensorflow/tensorflow/compiler/jit/
Dxla_device_context.cc64 tf_stats.bytes_in_use = se_stats->bytes_in_use; in GetStats()
/external/tensorflow/tensorflow/contrib/memory_stats/kernels/
Dmemory_stats_ops.cc54 return allocator_stats.bytes_in_use; in ExtractAllocatorStats()
/external/tensorflow/tensorflow/core/common_runtime/gpu/
Dgpu_bfc_allocator_test.cc38 static void CheckStats(Allocator* a, int64 num_allocs, int64 bytes_in_use, in CheckStats() argument
46 EXPECT_EQ(stats->bytes_in_use, bytes_in_use); in CheckStats()
/external/tensorflow/tensorflow/core/kernels/
Dstack.cc249 stats->bytes_in_use > (*stats->bytes_limit * kOccupancy)) { in ComputeAsync()
/external/tensorflow/tensorflow/core/common_runtime/eager/
Dkernel_and_device.cc191 memory->set_allocator_bytes_in_use(allocator_stats->bytes_in_use); in UpdateStats()
/external/e2fsprogs/lib/blkid/
Dprobe.h478 __u32 bytes_in_use; member