Searched refs:TrackingAllocator (Results 1 – 11 of 11) sorted by relevance
| /external/tensorflow/tensorflow/core/framework/ |
| D | tracking_allocator.cc | 23 TrackingAllocator::TrackingAllocator(Allocator* allocator, bool track_sizes) in TrackingAllocator() function in tensorflow::TrackingAllocator
32 void* TrackingAllocator::AllocateRaw( in AllocateRaw()
75 void TrackingAllocator::DeallocateRaw(void* ptr) { in DeallocateRaw()
112 bool TrackingAllocator::TracksAllocationSizes() const { in TracksAllocationSizes()
116 size_t TrackingAllocator::RequestedSize(const void* ptr) const { in RequestedSize()
129 size_t TrackingAllocator::AllocatedSize(const void* ptr) const { in AllocatedSize()
142 int64 TrackingAllocator::AllocationId(const void* ptr) const { in AllocationId()
155 absl::optional<AllocatorStats> TrackingAllocator::GetStats() { in GetStats()
159 void TrackingAllocator::ClearStats() { allocator_->ClearStats(); } in ClearStats()
161 std::tuple<size_t, size_t, size_t> TrackingAllocator::GetSizes() { in GetSizes()
[all …]
|
| D | tracking_allocator_test.cc | 72 TrackingAllocator* ta = new TrackingAllocator(a, false); in TEST()
90 ta = new TrackingAllocator(a, true); in TEST()
121 TrackingAllocator* ta = new TrackingAllocator(&a, false); in TEST()
147 TrackingAllocator* ta = new TrackingAllocator(&a, false); in TEST()
166 TrackingAllocator* ta = new TrackingAllocator(&a, false); in TEST()
|
| D | tracking_allocator.h | 54 class TrackingAllocator : public Allocator {
56 explicit TrackingAllocator(Allocator* allocator, bool track_ids);
91 ~TrackingAllocator() override {} in ~TrackingAllocator()
|
| D | allocator.cc | 76 cpu_alloc = new TrackingAllocator(cpu_alloc, true); in cpu_allocator_base()
|
| D | op_kernel.cc | 379 TrackingAllocator* wrapped_allocator = in get_allocator()
380 new TrackingAllocator(allocator, params_->track_allocations); in get_allocator()
|
| D | op_kernel.h | 590 typedef std::pair<Allocator*, TrackingAllocator*> WrappedAllocator;
|
| /external/flatbuffers/tests/rust_usage_test/bin/ |
| D | alloc_check.rs | 8 struct TrackingAllocator; struct
10 impl TrackingAllocator { implementation
15 unsafe impl GlobalAlloc for TrackingAllocator { implementation
27 static A: TrackingAllocator = TrackingAllocator;
|
| /external/tensorflow/tensorflow/core/common_runtime/ |
| D | step_stats_collector.h | 40 class TrackingAllocator; variable
124 TrackingAllocator* tracking_allocator);
126 gtl::InlinedVector<std::pair<AllocatorMemoryUsed*, TrackingAllocator*>, 2>
|
| D | process_state.cc | 123 allocator = new TrackingAllocator(allocator, true); in GetCPUAllocator()
|
| D | step_stats_collector.cc | 169 Allocator* allocator, TrackingAllocator* tracking_allocator) { in AddAllocation()
462 TrackingAllocator* tracking_alloc = alloc.second; in ReportAllocsOnResourceExhausted()
|
| /external/tensorflow/tensorflow/core/common_runtime/gpu/ |
| D | gpu_process_state.cc | 319 allocator = new TrackingAllocator(allocator, true); in GetGpuHostAllocator()
|