Searched defs:TrackingAllocator (Results 1 - 2 of 2) sorted by relevance

/external/tensorflow/tensorflow/core/framework/
H A Dtracking_allocator.h30 // TrackingAllocator is a wrapper for an Allocator. It keeps a running
33 // executions. Each Op gets a separate TrackingAllocator wrapper
44 // TrackingAllocator keeps track of outstanding calls using a
56 class TrackingAllocator : public Allocator { class in namespace:tensorflow
58 explicit TrackingAllocator(Allocator* allocator, bool track_ids);
93 ~TrackingAllocator() override {}
H A Dtracking_allocator.cc23 TrackingAllocator::TrackingAllocator(Allocator* allocator, bool track_sizes) function in class:tensorflow::TrackingAllocator
32 void* TrackingAllocator::AllocateRaw(
75 void TrackingAllocator::DeallocateRaw(void* ptr) {
112 bool TrackingAllocator::TracksAllocationSizes() {
116 size_t TrackingAllocator::RequestedSize(const void* ptr) {
129 size_t TrackingAllocator::AllocatedSize(const void* ptr) {
142 int64 TrackingAllocator::AllocationId(const void* ptr) {
155 void TrackingAllocator::GetStats(AllocatorStats* stats) {
159 void TrackingAllocator
[all...]

Completed in 110 milliseconds