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_process_state.cc172 Allocator* gpu_allocator = gpu_bfc_allocator; in GetGPUAllocator() local
183 gpu_allocator = new GPUDebugAllocator(gpu_allocator, platform_gpu_id); in GetGPUAllocator()
184 gpu_allocator = new GPUNanResetAllocator(gpu_allocator, platform_gpu_id); in GetGPUAllocator()
190 gpu_allocator = in GetGPUAllocator()
191 new GPUcudaMallocAllocator(gpu_allocator, platform_gpu_id); in GetGPUAllocator()
202 &process_state_->mem_desc_map_, gpu_allocator, md, &mu_); in GetGPUAllocator()
204 allocator_parts = {std::unique_ptr<Allocator>(gpu_allocator), in GetGPUAllocator()
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_device.h57 Allocator* gpu_allocator, Allocator* cpu_allocator,
363 const string& physical_device_desc, Allocator* gpu_allocator,
Dgpu_device.cc383 Allocator* gpu_allocator, Allocator* cpu_allocator, in BaseGPUDevice() argument
388 gpu_allocator_(gpu_allocator), in BaseGPUDevice()
1432 Allocator* gpu_allocator = process_state->GetGPUAllocator( in CreateGPUDevice() local
1434 if (gpu_allocator == nullptr) { in CreateGPUDevice()
1439 absl::optional<AllocatorStats> stats = gpu_allocator->GetStats(); in CreateGPUDevice()
1455 gpu_allocator, ProcessState::singleton()->GetCPUAllocator(numa_node)); in CreateGPUDevice()
/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()
/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()