| /external/tensorflow/tensorflow/core/framework/ |
| D | tracking_allocator_test.cc | 41 bool TracksAllocationSizes() override { return true; } in TracksAllocationSizes() function in tensorflow::TestableSizeTrackingAllocator
60 bool TracksAllocationSizes() 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() { in TracksAllocationSizes() function in tensorflow::TrackingAllocator
113 return track_sizes_locally_ || allocator_->TracksAllocationSizes(); in TracksAllocationSizes()
|
| D | allocator.h | 150 virtual bool TracksAllocationSizes() { return false; } in TracksAllocationSizes() function
200 if (TracksAllocationSizes()) { in AllocatedSizeSlow()
311 bool TracksAllocationSizes() override { in TracksAllocationSizes() function
312 return wrapped_->TracksAllocationSizes(); in TracksAllocationSizes()
316 return wrapped_->TracksAllocationSizes(); in ShouldAllocateEmptyTensors()
|
| D | tracking_allocator.h | 65 bool TracksAllocationSizes() override;
|
| D | allocator_test.cc | 152 EXPECT_EQ(false, a->TracksAllocationSizes()); in TEST()
|
| D | allocator.cc | 218 if (cpu_allocator_collect_full_stats && !cpu_alloc->TracksAllocationSizes()) { in cpu_allocator_base()
|
| D | op_kernel.cc | 750 if (a->TracksAllocationSizes()) { in allocate_temp()
773 if (a->TracksAllocationSizes()) { in allocate_persistent()
|
| D | tensor.cc | 82 if (alloc_->TracksAllocationSizes()) { in FillAllocationDescription()
|
| /external/tensorflow/tensorflow/core/common_runtime/ |
| D | process_state.h | 130 bool TracksAllocationSizes() override { return a_->TracksAllocationSizes(); } in TracksAllocationSizes() function
|
| D | scoped_allocator.h | 106 bool TracksAllocationSizes() override { return false; } in TracksAllocationSizes() function
|
| D | process_state.cc | 112 if (LogMemory::IsEnabled() && !allocator->TracksAllocationSizes()) { in GetCPUAllocator()
|
| D | bfc_allocator.h | 64 bool TracksAllocationSizes() override;
|
| D | bfc_allocator.cc | 525 bool BFCAllocator::TracksAllocationSizes() { return true; } in TracksAllocationSizes() function in tensorflow::BFCAllocator
|
| /external/tensorflow/tensorflow/core/common_runtime/gpu/ |
| D | gpu_cudamalloc_allocator.h | 40 bool TracksAllocationSizes() override;
|
| D | gpu_cudamalloc_allocator.cc | 64 bool GPUcudaMallocAllocator::TracksAllocationSizes() { return false; } in TracksAllocationSizes() function in tensorflow::GPUcudaMallocAllocator
|
| D | gpu_debug_allocator.h | 42 bool TracksAllocationSizes() override;
|
| D | gpu_debug_allocator.cc | 117 bool GPUDebugAllocator::TracksAllocationSizes() { return true; } in TracksAllocationSizes() function in tensorflow::GPUDebugAllocator
|
| D | gpu_debug_allocator_test.cc | 231 EXPECT_EQ(true, a.TracksAllocationSizes()); in TEST()
|
| D | gpu_process_state.cc | 247 if (LogMemory::IsEnabled() && !allocator->TracksAllocationSizes()) { in GetGpuHostAllocator()
|
| D | gpu_bfc_allocator_test.cc | 202 EXPECT_EQ(true, a.TracksAllocationSizes()); in TEST()
|
| /external/tensorflow/tensorflow/core/common_runtime/sycl/ |
| D | sycl_allocator.h | 50 bool TracksAllocationSizes() override { return true; } in TracksAllocationSizes() function
|
| /external/tensorflow/tensorflow/core/grappler/clusters/ |
| D | single_machine.cc | 231 if (!allocator->TracksAllocationSizes()) { in GetPeakMemoryUsage()
459 if (!allocator->TracksAllocationSizes()) { in ClearAllocatorStats()
|