Searched refs:gpu_allocator (Results 1 – 5 of 5) sorted by relevance
| /external/tensorflow/tensorflow/core/common_runtime/gpu/ |
| D | gpu_process_state.cc | 117 Allocator* gpu_allocator = gpu_bfc_allocator; in GetGPUAllocator() local
127 gpu_allocator = new GPUDebugAllocator(gpu_allocator, platform_gpu_id); in GetGPUAllocator()
128 gpu_allocator = new GPUNanResetAllocator(gpu_allocator, platform_gpu_id); in GetGPUAllocator()
133 gpu_allocator = in GetGPUAllocator()
134 new GPUcudaMallocAllocator(gpu_allocator, platform_gpu_id); in GetGPUAllocator()
145 &process_state_->mem_desc_map_, gpu_allocator, md, &mu_); in GetGPUAllocator()
147 allocator_parts = {std::unique_ptr<Allocator>(gpu_allocator), in GetGPUAllocator()
|
| D | gpu_device_factory.cc | 33 Allocator* gpu_allocator, Allocator* cpu_allocator) in GPUDevice() argument
35 physical_device_desc, gpu_allocator, cpu_allocator, in GPUDevice()
66 const string& physical_device_desc, Allocator* gpu_allocator, in CreateGPUDevice() argument
70 gpu_allocator, cpu_allocator); in CreateGPUDevice()
|
| D | gpu_device.h | 57 Allocator* gpu_allocator, Allocator* cpu_allocator,
308 const string& physical_device_desc, Allocator* gpu_allocator,
|
| D | gpu_device.cc | 302 Allocator* gpu_allocator, Allocator* cpu_allocator, in BaseGPUDevice() argument
307 gpu_allocator_(gpu_allocator), in BaseGPUDevice()
1199 Allocator* gpu_allocator = process_state->GetGPUAllocator( in CreateGPUDevice() local
1201 if (gpu_allocator == nullptr) { in CreateGPUDevice()
1206 absl::optional<AllocatorStats> stats = gpu_allocator->GetStats(); in CreateGPUDevice()
1221 gpu_allocator, ProcessState::singleton()->GetCPUAllocator(numa_node)); in CreateGPUDevice()
|
| /external/tensorflow/tensorflow/core/kernels/ |
| D | stack.cc | 304 Allocator* gpu_allocator = device->GetAllocator(value.alloc_attrs); in ComputeAsync() local
306 new Tensor(gpu_allocator, cpu_tensor->dtype(), cpu_tensor->shape()); in ComputeAsync()
|