| /external/tensorflow/tensorflow/core/framework/ |
| D | tracking_allocator_test.cc | 41 bool TracksAllocationSizes() const override { return true; } in TracksAllocationSizes() function in tensorflow::TestableSizeTrackingAllocator
60 bool TracksAllocationSizes() const override { return true; } in TracksAllocationSizes() function in tensorflow::NoMemoryAllocator
67 EXPECT_FALSE(a->TracksAllocationSizes()); in TEST()
119 EXPECT_TRUE(a.TracksAllocationSizes()); in TEST()
145 EXPECT_TRUE(a.TracksAllocationSizes()); in TEST()
164 EXPECT_TRUE(a.TracksAllocationSizes()); in TEST()
|
| D | tracking_allocator.cc | 29 track_sizes_locally_(track_sizes && !allocator_->TracksAllocationSizes()), in TrackingAllocator()
41 if (allocator_->TracksAllocationSizes()) { in AllocateRaw()
83 bool tracks_allocation_sizes = allocator_->TracksAllocationSizes(); in DeallocateRaw()
112 bool TrackingAllocator::TracksAllocationSizes() const { in TracksAllocationSizes() function in tensorflow::TrackingAllocator
113 return track_sizes_locally_ || allocator_->TracksAllocationSizes(); in TracksAllocationSizes()
|
| D | allocator.h | 156 virtual bool TracksAllocationSizes() const { return false; } in TracksAllocationSizes() function
217 if (TracksAllocationSizes()) { in AllocatedSizeSlow()
258 bool TracksAllocationSizes() const override { in TracksAllocationSizes() function
259 return wrapped_->TracksAllocationSizes(); in TracksAllocationSizes()
|
| D | allocator.cc | 74 if (cpu_allocator_collect_full_stats && !cpu_alloc->TracksAllocationSizes()) { in cpu_allocator_base()
|
| D | tracking_allocator.h | 64 bool TracksAllocationSizes() const override;
|
| D | allocator_test.cc | 201 EXPECT_EQ(false, a->TracksAllocationSizes()); in TEST()
|
| D | tensor.cc | 90 if (alloc_->TracksAllocationSizes()) { in GetAllocatedBytes()
104 if (alloc_->TracksAllocationSizes()) { in FillAllocationDescription()
|
| D | op_kernel.cc | 840 if (a->TracksAllocationSizes()) { in allocate_temp()
876 if (a->TracksAllocationSizes()) { in allocate_persistent()
|
| /external/tensorflow/tensorflow/core/common_runtime/ |
| D | process_state.h | 131 bool TracksAllocationSizes() const override { in TracksAllocationSizes() function
132 return a_->TracksAllocationSizes(); in TracksAllocationSizes()
|
| D | scoped_allocator.h | 109 bool TracksAllocationSizes() const override { return false; } in TracksAllocationSizes() function
|
| D | process_state.cc | 112 if (LogMemory::IsEnabled() && !allocator->TracksAllocationSizes()) { in GetCPUAllocator()
|
| D | bfc_allocator.h | 68 bool TracksAllocationSizes() const override;
|
| D | bfc_allocator.cc | 819 bool BFCAllocator::TracksAllocationSizes() const { return true; } in TracksAllocationSizes() function in tensorflow::BFCAllocator
|
| /external/tensorflow/tensorflow/core/common_runtime/gpu/ |
| D | gpu_cudamalloc_allocator.h | 40 bool TracksAllocationSizes() const override;
|
| D | gpu_cudamalloc_allocator.cc | 68 bool GPUcudaMallocAllocator::TracksAllocationSizes() const { return false; } in TracksAllocationSizes() function in tensorflow::GPUcudaMallocAllocator
|
| D | gpu_debug_allocator.h | 42 bool TracksAllocationSizes() const override;
|
| D | gpu_debug_allocator.cc | 117 bool GPUDebugAllocator::TracksAllocationSizes() const { return true; } in TracksAllocationSizes() function in tensorflow::GPUDebugAllocator
|
| D | gpu_debug_allocator_test.cc | 236 EXPECT_EQ(true, a.TracksAllocationSizes()); in TEST()
|
| D | gpu_process_state.cc | 258 if (LogMemory::IsEnabled() && !allocator->TracksAllocationSizes()) { in GetGpuHostAllocator()
|
| D | gpu_bfc_allocator_test.cc | 205 EXPECT_EQ(true, a.TracksAllocationSizes()); in TEST()
|
| /external/tensorflow/tensorflow/core/common_runtime/sycl/ |
| D | sycl_allocator.h | 50 bool TracksAllocationSizes() const override { return true; } in TracksAllocationSizes() function
|
| /external/tensorflow/tensorflow/core/grappler/clusters/ |
| D | single_machine.cc | 228 if (!allocator->TracksAllocationSizes()) { in GetPeakMemoryUsage()
456 if (!allocator->TracksAllocationSizes()) { in ClearAllocatorStats()
|