Searched refs:gpu_allocator_ (Results 1 – 4 of 4) sorted by relevance
54 return gpu_allocator_; in GetAllocator()
388 gpu_allocator_(gpu_allocator), in BaseGPUDevice()398 gpu_allocator_->DeallocateRaw(scratch_); in ~BaseGPUDevice()409 void* scratch_buffer = gpu_allocator_->AllocateRaw( in InitScratchBuffers()469 timestamped_allocator_ ? gpu_allocator_ : nullptr, em_)); in Init()1074 return gpu_allocator_; in GetScopedAllocator()
123 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