Searched defs:TracksAllocationSizes (Results 1 - 5 of 5) sorted by relevance

/external/tensorflow/tensorflow/core/common_runtime/gpu/
H A Dgpu_cudamalloc_allocator.cc71 bool GPUcudaMallocAllocator::TracksAllocationSizes() { return false; } function in class:tensorflow::GPUcudaMallocAllocator
H A Dgpu_debug_allocator.cc122 bool GPUDebugAllocator::TracksAllocationSizes() { return true; } function in class:tensorflow::GPUDebugAllocator
/external/tensorflow/tensorflow/core/framework/
H A Dtracking_allocator.cc29 track_sizes_locally_(track_sizes && !allocator_->TracksAllocationSizes()),
41 if (allocator_->TracksAllocationSizes()) {
83 bool tracks_allocation_sizes = allocator_->TracksAllocationSizes();
112 bool TrackingAllocator::TracksAllocationSizes() { function in class:tensorflow::TrackingAllocator
113 return track_sizes_locally_ || allocator_->TracksAllocationSizes();
H A Dallocator.h141 // TracksAllocationSizes is overridden to return true.
142 virtual bool TracksAllocationSizes() { return false; } function in class:tensorflow::Allocator
155 // REQUIRES: TracksAllocationSizes() is true.
168 // REQUIRES: TracksAllocationSizes() is true.
179 // REQUIRES: TracksAllocationSizes() is true.
187 // TracksAllocationSizes() is false, but can be extremely slow.
192 if (TracksAllocationSizes()) {
307 bool TracksAllocationSizes() override {
308 return wrapped_->TracksAllocationSizes();
312 return wrapped_->TracksAllocationSizes();
[all...]
/external/tensorflow/tensorflow/core/common_runtime/
H A Dbfc_allocator.cc522 bool BFCAllocator::TracksAllocationSizes() { return true; } function in class:BFCAllocator

Completed in 121 milliseconds