Searched refs:tracking_allocator (Results 1 – 4 of 4) sorted by relevance
| /external/tensorflow/tensorflow/core/common_runtime/ |
| D | step_stats_collector.cc | 169 Allocator* allocator, TrackingAllocator* tracking_allocator) { in AddAllocation() argument
172 auto sizes = tracking_allocator->GetSizes(); in AddAllocation()
181 allocations_.push_back(std::make_pair(memory, tracking_allocator)); in AddAllocation()
|
| D | step_stats_collector.h | 124 TrackingAllocator* tracking_allocator);
|
| /external/tensorflow/tensorflow/core/framework/ |
| D | BUILD | 94 "tracking_allocator.h",
243 "tracking_allocator.h",
321 "tracking_allocator.cc",
322 "tracking_allocator.h",
472 "tracking_allocator.cc",
473 "tracking_allocator.h",
518 "tracking_allocator.h",
|
| /external/tensorflow/tensorflow/core/ |
| D | BUILD | 466 "//tensorflow/core/framework:tracking_allocator.h",
1546 "//tensorflow/core/framework:tracking_allocator.h", # only needed for tests
|