Searched refs:gpu_allocator_ (Results 1 – 4 of 4) sorted by relevance
54 return gpu_allocator_; in GetAllocator()
390 gpu_allocator_(gpu_allocator), in BaseGPUDevice()400 gpu_allocator_->DeallocateRaw(scratch_); in ~BaseGPUDevice()411 void* scratch_buffer = gpu_allocator_->AllocateRaw( in InitScratchBuffers()472 timestamped_allocator_ ? gpu_allocator_ : nullptr, em_)); in Init()1131 return gpu_allocator_; in GetScopedAllocator()
129 Allocator* gpu_allocator_; // not owned
151 gpu_allocator_ = GPUProcessState::singleton()->GetGPUAllocator( in GPUDeviceTestHelper()157 Allocator* gpu_allocator() { return gpu_allocator_; } in gpu_allocator()169 Allocator* gpu_allocator_; member in tensorflow::GPUDeviceTestHelper