Home
last modified time | relevance | path

Searched refs:GetGPUAllocator (Results 1 – 5 of 5) sorted by relevance

/external/tensorflow/tensorflow/core/common_runtime/gpu/
Dgpu_process_state.h84 virtual Allocator* GetGPUAllocator(const GPUOptions& options,
Dgpu_process_state.cc136 Allocator* GPUProcessState::GetGPUAllocator( in GetGPUAllocator() function in tensorflow::GPUProcessState
Dgpu_device.cc1432 Allocator* gpu_allocator = process_state->GetGPUAllocator( in CreateGPUDevice()
/external/tensorflow/tensorflow/core/common_runtime/device/
Ddevice_event_mgr_test.cc151 gpu_allocator_ = GPUProcessState::singleton()->GetGPUAllocator( in GPUDeviceTestHelper()
/external/tensorflow/tensorflow/compiler/tf2tensorrt/convert/
Dconvert_graph.cc647 dev_allocator = GPUProcessState::singleton()->GetGPUAllocator( in GetDeviceAndAllocator()