HomeSort by relevance Sort by last modified time
    Searched defs:GPUDebugAllocator (Results 1 - 2 of 2) sorted by null

  /external/tensorflow/tensorflow/core/common_runtime/gpu/
gpu_debug_allocator.h 34 class GPUDebugAllocator : public VisitableAllocator {
36 explicit GPUDebugAllocator(VisitableAllocator* allocator,
38 ~GPUDebugAllocator() override;
60 TF_DISALLOW_COPY_AND_ASSIGN(GPUDebugAllocator);
gpu_debug_allocator.cc 76 // GPUDebugAllocator
78 GPUDebugAllocator::GPUDebugAllocator(VisitableAllocator* allocator,
84 GPUDebugAllocator::~GPUDebugAllocator() { delete base_allocator_; }
86 void* GPUDebugAllocator::AllocateRaw(size_t alignment, size_t num_bytes) {
104 void GPUDebugAllocator::DeallocateRaw(void* ptr) {
114 void GPUDebugAllocator::AddAllocVisitor(Visitor visitor) {
118 void GPUDebugAllocator::AddFreeVisitor(Visitor visitor) {
122 bool GPUDebugAllocator::TracksAllocationSizes() { return true;
    [all...]

Completed in 120 milliseconds