| /external/tensorflow/tensorflow/contrib/memory_stats/kernels/ |
| D | memory_stats_ops.cc | 27 absl::optional<AllocatorStats> allocator_stats = allocator->GetStats(); in Compute()
29 *allocator_stats = AllocatorStats(); in Compute()
42 const AllocatorStats& allocator_stats) const = 0;
53 const AllocatorStats& allocator_stats) const override { in ExtractAllocatorStats()
75 const AllocatorStats& allocator_stats) const override { in ExtractAllocatorStats()
97 const AllocatorStats& allocator_stats) const override { in ExtractAllocatorStats()
|
| /external/tensorflow/tensorflow/stream_executor/ |
| D | allocator_stats.h | 29 struct AllocatorStats { struct
39 AllocatorStats() in AllocatorStats() argument
|
| D | allocator_stats.cc | 21 string AllocatorStats::DebugString() const { in DebugString()
|
| D | stream_executor_internal.h | 373 virtual absl::optional<AllocatorStats> GetAllocatorStats() { in GetAllocatorStats()
|
| /external/tensorflow/tensorflow/core/common_runtime/ |
| D | mkl_cpu_allocator.h | 85 absl::optional<AllocatorStats> GetStats() override { in GetStats()
126 AllocatorStats stats_ GUARDED_BY(mutex_);
251 absl::optional<AllocatorStats> GetStats() override { in GetStats()
315 AllocatorStats stats_ GUARDED_BY(mutex_);
|
| D | process_state.h | 133 absl::optional<AllocatorStats> GetStats() override { return a_->GetStats(); } in GetStats()
|
| D | bfc_allocator.h | 72 absl::optional<AllocatorStats> GetStats() override;
456 AllocatorStats stats_ GUARDED_BY(lock_);
|
| /external/compiler-rt/lib/sanitizer_common/ |
| D | sanitizer_allocator.h | 209 class AllocatorStats {
236 AllocatorStats *next_;
237 AllocatorStats *prev_;
242 class AllocatorGlobalStats : public AllocatorStats {
253 void Register(AllocatorStats *s) { in Register()
261 void Unregister(AllocatorStats *s) { in Unregister()
272 const AllocatorStats *stats = this; in Get()
353 NOINLINE Batch* AllocateBatch(AllocatorStats *stat, AllocatorCache *c, in AllocateBatch()
364 NOINLINE void DeallocateBatch(AllocatorStats *stat, uptr class_id, Batch *b) { in DeallocateBatch()
538 NOINLINE Batch* PopulateFreeList(AllocatorStats *stat, AllocatorCache *c, in PopulateFreeList()
[all …]
|
| /external/tensorflow/tensorflow/core/common_runtime/sycl/ |
| D | sycl_allocator.h | 46 void GetStats(AllocatorStats* stats) override;
68 AllocatorStats stats_ GUARDED_BY(mu_);
|
| D | sycl_allocator.cc | 69 void SYCLAllocator::GetStats(AllocatorStats* stats) { in GetStats()
|
| /external/tensorflow/tensorflow/core/common_runtime/gpu/ |
| D | gpu_debug_allocator.h | 46 absl::optional<AllocatorStats> GetStats() override;
74 absl::optional<AllocatorStats> GetStats() override;
|
| D | gpu_debug_allocator.cc | 135 absl::optional<AllocatorStats> GPUDebugAllocator::GetStats() { in GetStats()
211 absl::optional<AllocatorStats> GPUNanResetAllocator::GetStats() { in GetStats()
|
| D | gpu_bfc_allocator_test.cc | 40 absl::optional<AllocatorStats> stats = a->GetStats(); in CheckStats()
297 absl::optional<AllocatorStats> stats = a.GetStats(); in TEST()
|
| /external/tensorflow/tensorflow/core/framework/ |
| D | allocator.cc | 29 string AllocatorStats::DebugString() const { in DebugString()
153 absl::optional<AllocatorStats> GetStats() override { in GetStats()
171 AllocatorStats stats_ GUARDED_BY(mu_);
|
| D | allocator.h | 60 struct AllocatorStats { struct
70 AllocatorStats() in AllocatorStats() argument
207 virtual absl::optional<AllocatorStats> GetStats() { return absl::nullopt; } in GetStats()
|
| D | tracking_allocator_test.cc | 47 absl::optional<AllocatorStats> GetStats() override { return absl::nullopt; } in GetStats()
61 absl::optional<AllocatorStats> GetStats() override { return absl::nullopt; } in GetStats()
|
| D | tracking_allocator.h | 69 absl::optional<AllocatorStats> GetStats() override;
|
| D | tracking_allocator.cc | 155 absl::optional<AllocatorStats> TrackingAllocator::GetStats() { in GetStats()
|
| D | allocator_test.cc | 29 absl::optional<AllocatorStats> stats = a->GetStats(); in CheckStats()
|
| /external/tensorflow/tensorflow/compiler/jit/ |
| D | xla_device_context.cc | 55 absl::optional<AllocatorStats> XlaDeviceAllocator::GetStats() { in GetStats()
56 absl::optional<stream_executor::AllocatorStats> se_stats = in GetStats()
62 tensorflow::AllocatorStats tf_stats; in GetStats()
|
| D | xla_device_context.h | 44 absl::optional<AllocatorStats> GetStats() override;
|
| /external/compiler-rt/lib/sanitizer_common/tests/ |
| D | sanitizer_allocator_test.cc | 270 AllocatorStats stats; in TEST()
297 AllocatorStats stats; in TEST()
314 AllocatorStats stats; in TEST()
329 AllocatorStats stats; in FailInAssertionOnOOM()
347 AllocatorStats stats; in TEST()
718 AllocatorStats stats; in TEST()
745 AllocatorStats stats; in TEST()
|
| /external/tensorflow/tensorflow/core/kernels/ |
| D | stack.cc | 247 absl::optional<AllocatorStats> stats = allocator->GetStats(); in ComputeAsync()
|
| /external/tensorflow/tensorflow/core/grappler/clusters/ |
| D | single_machine.cc | 235 absl::optional<AllocatorStats> stats = allocator->GetStats(); in GetPeakMemoryUsage()
|
| /external/tensorflow/tensorflow/core/common_runtime/eager/ |
| D | kernel_and_device.cc | 188 absl::optional<AllocatorStats> allocator_stats = in UpdateStats()
|