Home
last modified time | relevance | path

Searched refs:gpu_allocator (Results 1 – 6 of 6) sorted by relevance

/external/tensorflow/tensorflow/core/common_runtime/gpu/
Dgpu_device_factory.cc34 Allocator* gpu_allocator, Allocator* cpu_allocator) in GPUDevice() argument
36 physical_device_desc, gpu_allocator, cpu_allocator, in GPUDevice()
67 const string& physical_device_desc, Allocator* gpu_allocator, in CreateGPUDevice() argument
71 gpu_allocator, cpu_allocator); in CreateGPUDevice()
Dgpu_process_state.cc198 Allocator* gpu_allocator = gpu_bfc_allocator.get(); in GetGPUAllocator() local
210 gpu_allocator = new GPUNanResetAllocator( in GetGPUAllocator()
211 new GPUDebugAllocator(gpu_allocator, platform_device_id), in GetGPUAllocator()
219 gpu_allocator = new GPUcudaMallocAllocator(platform_device_id); in GetGPUAllocator()
229 gpu_allocator = in GetGPUAllocator()
241 &process_state_->mem_desc_map_, gpu_allocator, md, &mu_); in GetGPUAllocator()
244 std::unique_ptr<Allocator>(gpu_allocator), in GetGPUAllocator()
Dgpu_device.h81 Allocator* gpu_allocator, Allocator* cpu_allocator,
398 const string& physical_device_desc, Allocator* gpu_allocator,
Dgpu_device.cc391 Allocator* gpu_allocator, Allocator* cpu_allocator, in BaseGPUDevice() argument
396 gpu_allocator_(gpu_allocator), in BaseGPUDevice()
1592 Allocator* gpu_allocator = process_state->GetGPUAllocator( in CreateGPUDevice() local
1594 if (gpu_allocator == nullptr) { in CreateGPUDevice()
1599 absl::optional<AllocatorStats> stats = gpu_allocator->GetStats(); in CreateGPUDevice()
1615 gpu_allocator, ProcessState::singleton()->GetCPUAllocator(numa_node)); in CreateGPUDevice()
1620 gpu_allocator->SetStreamAndPreallocateMemory(gpu_device->GetStream()); in CreateGPUDevice()
/external/tensorflow/tensorflow/core/kernels/
Dstack.cc305 Allocator* gpu_allocator = device->GetAllocator(value.alloc_attrs); in ComputeAsync() local
307 new Tensor(gpu_allocator, cpu_tensor->dtype(), cpu_tensor->shape()); in ComputeAsync()
/external/tensorflow/tensorflow/core/common_runtime/device/
Ddevice_event_mgr_test.cc157 Allocator* gpu_allocator() { return gpu_allocator_; } in gpu_allocator() function in tensorflow::GPUDeviceTestHelper
209 gpu_inputs_.push_back(Tensor(gpu_helper_->gpu_allocator(), DT_FLOAT, in ReInit()
214 gpu_outputs_.push_back(Tensor(gpu_helper_->gpu_allocator(), DT_FLOAT, in ReInit()
245 gpu_helper_->gpu_allocator(), node_def, in GetOpKernel()