Searched refs:max_bytes_in_use (Results 1 – 11 of 11) sorted by relevance
| /external/tensorflow/tensorflow/contrib/memory_stats/python/kernel_tests/ |
| D | memory_stats_ops_test.py | 69 max_bytes_in_use = sess.run(max_bytes_in_use_op)
70 self.assertGreaterEqual(max_bytes_in_use, matrix_size_in_bytes * 3)
71 self.assertLess(max_bytes_in_use, matrix_size_in_bytes * 4)
82 _, bytes_in_use, max_bytes_in_use = sess.run([c, bytes_in_use_op,
90 self.assertGreaterEqual(max_bytes_in_use, matrix_size_in_bytes * 3)
|
| /external/tensorflow/tensorflow/core/framework/ |
| D | allocator.cc | 31 this->max_bytes_in_use = 0; in Clear()
43 this->bytes_limit, this->bytes_in_use, this->max_bytes_in_use, in DebugString()
86 stats_.max_bytes_in_use = in AllocateRaw()
87 std::max<int64>(stats_.max_bytes_in_use, stats_.bytes_in_use); in AllocateRaw()
112 stats_.max_bytes_in_use = stats_.bytes_in_use; in ClearStats()
|
| D | allocator_test.cc | 28 int64 max_bytes_in_use, int64 max_alloc_size) { in CheckStats() argument
38 EXPECT_GT(stats.max_bytes_in_use, max_bytes_in_use - kSlop); in CheckStats()
39 EXPECT_LT(stats.max_bytes_in_use, max_bytes_in_use + kSlop); in CheckStats()
|
| D | allocator.h | 52 int64 max_bytes_in_use; // The maximum bytes in use. member
|
| /external/tensorflow/tensorflow/core/common_runtime/sycl/ |
| D | sycl_allocator.cc | 51 stats_.max_bytes_in_use = in AllocateRaw()
52 std::max<int64>(stats_.max_bytes_in_use, stats_.bytes_in_use); in AllocateRaw()
77 stats_.max_bytes_in_use = stats_.bytes_in_use; in ClearStats()
|
| /external/tensorflow/tensorflow/core/profiler/internal/ |
| D | tfprof_timeline.cc | 278 int64 max_bytes_in_use = 0; in GenerateGraphTimeline() local
283 max_bytes_in_use = std::max(max_bytes_in_use, cur_bytes_in_use); in GenerateGraphTimeline()
305 max_bytes_in_use / 1000000.0); in GenerateGraphTimeline()
|
| /external/tensorflow/tensorflow/contrib/memory_stats/python/ops/ |
| D | memory_stats_ops.py | 41 return gen_memory_stats_ops.max_bytes_in_use()
|
| /external/tensorflow/tensorflow/core/common_runtime/gpu/ |
| D | gpu_bfc_allocator_test.cc | 38 int64 max_bytes_in_use, int64 max_alloc_size) { in CheckStats() argument
43 EXPECT_EQ(stats.max_bytes_in_use, max_bytes_in_use); in CheckStats()
|
| /external/tensorflow/tensorflow/contrib/memory_stats/kernels/ |
| D | memory_stats_ops.cc | 96 return allocator_stats.max_bytes_in_use; in ExtractAllocatorStats()
|
| /external/tensorflow/tensorflow/core/common_runtime/ |
| D | bfc_allocator.cc | 322 stats_.max_bytes_in_use = in FindChunkPtr()
323 std::max(stats_.max_bytes_in_use, stats_.bytes_in_use); in FindChunkPtr()
691 stats_.max_bytes_in_use = stats_.bytes_in_use; in ClearStats()
|
| /external/tensorflow/tensorflow/core/grappler/clusters/ |
| D | single_machine.cc | 224 (*device_peak_memory)[device->name()] = stats.max_bytes_in_use; in GetPeakMemoryUsage()
|