Home
last modified time | relevance | path

Searched defs:peer_gpu_ids (Results 1 – 3 of 3) sorted by relevance

/external/tensorflow/tensorflow/core/common_runtime/gpu/
Dgpu_process_state.cc86 size_t total_bytes, const std::vector<TfGpuId>& peer_gpu_ids) { in CreateSubAllocator()
138 const std::vector<TfGpuId>& peer_gpu_ids) { in GetGPUAllocator()
Dgpu_virtual_mem_allocator.cc66 const std::vector<PlatformGpuId>& peer_gpu_ids) { in Create()
Dgpu_device.cc1422 std::vector<TfGpuId> peer_gpu_ids; in CreateGPUDevice() local