| /external/tensorflow/tensorflow/core/common_runtime/ |
| D | mkl_cpu_allocator.h | 93 stats_.peak_bytes_in_use = 0; in ClearStats()
105 stats_.peak_bytes_in_use = in IncrementStats()
106 std::max(stats_.peak_bytes_in_use, stats_.bytes_in_use); in IncrementStats()
260 stats_.peak_bytes_in_use = in GetStats()
261 l_stats->peak_bytes_in_use + s_stats->peak_bytes_in_use; in GetStats()
|
| D | bfc_allocator.cc | 510 {"peak_bytes_in_use", stats_.peak_bytes_in_use}, in AddTraceMe()
564 stats_.peak_bytes_in_use = in FindChunkPtr()
565 std::max(stats_.peak_bytes_in_use, stats_.bytes_in_use); in FindChunkPtr()
1099 mas->set_peak_bytes_in_use(stats_.peak_bytes_in_use); in RecordMemoryMapInternal()
1160 stats_.peak_bytes_in_use = stats_.bytes_in_use; in ClearStats()
|
| /external/tensorflow/tensorflow/core/framework/ |
| D | allocator_test.cc | 29 int64 peak_bytes_in_use, int64 largest_alloc_size) { in CheckStats() argument
42 EXPECT_GT(stats->peak_bytes_in_use, peak_bytes_in_use - kSlop); in CheckStats()
43 EXPECT_LT(stats->peak_bytes_in_use, peak_bytes_in_use + kSlop); in CheckStats()
|
| D | cpu_allocator_impl.cc | 91 stats_.peak_bytes_in_use = in AllocateRaw()
92 std::max<int64>(stats_.peak_bytes_in_use, stats_.bytes_in_use); in AllocateRaw()
125 stats_.peak_bytes_in_use = stats_.bytes_in_use; in ClearStats()
|
| D | allocator.cc | 44 static_cast<long long>(this->peak_bytes_in_use), in DebugString()
|
| D | allocator.h | 151 int64 peak_bytes_in_use; // The peak bytes in use. member
170 peak_bytes_in_use(0), in AllocatorStats()
|
| /external/tensorflow/tensorflow/stream_executor/ |
| D | allocator_stats.h | 32 int64 peak_bytes_in_use; // The peak bytes in use. member
51 peak_bytes_in_use(0), in AllocatorStats()
|
| D | allocator_stats.cc | 32 this->peak_bytes_in_use, this->num_allocs, this->largest_alloc_size, in DebugString()
|
| /external/tensorflow/tensorflow/core/protobuf/ |
| D | bfc_memory_map.proto | 11 int64 peak_bytes_in_use = 3; field
|
| /external/tensorflow/tensorflow/compiler/jit/ |
| D | xla_device_context.cc | 66 tf_stats.peak_bytes_in_use = se_stats->peak_bytes_in_use; in GetStats()
|
| /external/tensorflow/tensorflow/core/profiler/convert/ |
| D | xplane_to_memory_profile.cc | 65 summary->set_peak_bytes_usage_lifetime(stats.peak_bytes_in_use()); in UpdateProfileSummary()
70 peak_stats->peak_bytes_in_use()) { in UpdateProfileSummary()
509 .peak_bytes_in_use(), in ProcessMemoryProfileProto()
|
| D | xplane_to_memory_profile_test.cc | 102 .peak_bytes_in_use(), in TEST()
|
| /external/tensorflow/tensorflow/stream_executor/tpu/ |
| D | tpu_executor.cc | 231 stats.peak_bytes_in_use = c_stats.peak_bytes_in_use; in GetAllocatorStats()
|
| D | c_api_decl.h | 85 int64_t peak_bytes_in_use; member
|
| /external/tensorflow/tensorflow/core/common_runtime/gpu/ |
| D | gpu_bfc_allocator_test.cc | 46 int64 peak_bytes_in_use, int64 largest_alloc_size) { in CheckStats() argument
54 EXPECT_EQ(stats->peak_bytes_in_use, peak_bytes_in_use); in CheckStats()
|
| /external/tensorflow/tensorflow/c/experimental/stream_executor/ |
| D | stream_executor.h | 111 int64_t peak_bytes_in_use; member
|
| D | stream_executor.cc | 319 stats.peak_bytes_in_use = c_stats.peak_bytes_in_use; in GetAllocatorStats()
|
| /external/tensorflow/tensorflow/core/profiler/protobuf/ |
| D | memory_profile.proto | 32 int64 peak_bytes_in_use = 5; field
|
| /external/tensorflow/tensorflow/core/debug/ |
| D | bfc_dump_reader.cc | 110 static_cast<int64_t>(md.stats().peak_bytes_in_use()), in PrintSummary()
|
| /external/tensorflow/tensorflow/core/grappler/clusters/ |
| D | single_machine.cc | 234 (stats ? stats->peak_bytes_in_use : 0); in GetPeakMemoryUsage()
|
| /external/tensorflow/tensorflow/python/ |
| D | tfe_wrapper.cc | 581 {"peak", stats->peak_bytes_in_use}}; in PYBIND11_MODULE()
|