Searched refs:gpu_allocator (Results 1 – 9 of 9) 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
128 gpu_allocator = new GPUDebugAllocator(gpu_allocator, platform_gpu_id); in GetGPUAllocator()
129 gpu_allocator = new GPUNanResetAllocator(gpu_allocator, platform_gpu_id); in GetGPUAllocator()
135 gpu_allocator = in GetGPUAllocator()
136 new GPUcudaMallocAllocator(gpu_allocator, platform_gpu_id); in GetGPUAllocator()
147 &process_state_->mem_desc_map_, gpu_allocator, md, &mu_); in GetGPUAllocator()
149 allocator_parts = {std::unique_ptr<Allocator>(gpu_allocator), in GetGPUAllocator()
|
| D | gpu_device_factory.cc | 34 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()
|
| D | gpu_device.h | 57 Allocator* gpu_allocator, Allocator* cpu_allocator,
362 const string& physical_device_desc, Allocator* gpu_allocator,
|
| D | gpu_event_mgr_test.cc | 302 Allocator* gpu_allocator() { return gpu_allocator_; } in gpu_allocator() function in tensorflow::GPUDeviceTestHelper
354 gpu_inputs_.push_back(Tensor(gpu_helper_->gpu_allocator(), DT_FLOAT, in ReInit()
359 gpu_outputs_.push_back(Tensor(gpu_helper_->gpu_allocator(), DT_FLOAT, in ReInit()
390 gpu_helper_->gpu_allocator(), node_def, in GetOpKernel()
|
| D | gpu_device.cc | 318 Allocator* gpu_allocator, Allocator* cpu_allocator, in BaseGPUDevice() argument
323 gpu_allocator_(gpu_allocator), in BaseGPUDevice()
1220 Allocator* gpu_allocator = process_state->GetGPUAllocator( in CreateGPUDevice() local
1222 if (gpu_allocator == nullptr) { in CreateGPUDevice()
1227 absl::optional<AllocatorStats> stats = gpu_allocator->GetStats(); in CreateGPUDevice()
1243 gpu_allocator, ProcessState::singleton()->GetCPUAllocator(numa_node)); in CreateGPUDevice()
|
| /external/crosvm/resources/src/ |
| D | system_allocator.rs | 8 use crate::gpu_allocator::{self, GpuMemoryAllocator};
44 gpu_allocator: Option<Box<dyn GpuMemoryAllocator>>, field
81 gpu_allocator: if create_gpu_allocator { in new()
82 gpu_allocator::create_gpu_memory_allocator().map_err(Error::CreateGpuAllocator)? in new()
123 self.gpu_allocator.as_ref().map(|v| v.as_ref()) in gpu_memory_allocator()
|
| D | lib.rs | 16 pub use crate::gpu_allocator::{
22 mod gpu_allocator; module
|
| /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()
|
| /external/crosvm/vm_control/src/ |
| D | lib.rs | 230 Some(gpu_allocator) => match gpu_allocator.allocate(width, height, format) { in execute()
|