Home
last modified time | relevance | path

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

/aosp_15_r20/external/tensorflow/tensorflow/core/common_runtime/gpu/
H A Dgpu_process_state.cc101 size_t total_bytes, const std::vector<TfDeviceId>& peer_gpu_ids) { in CreateSubAllocator()
153 const std::vector<TfDeviceId>& peer_gpu_ids) { in GetGPUAllocator()
H A Dgpu_virtual_mem_allocator.cc66 const std::vector<PlatformDeviceId>& peer_gpu_ids) { in Create()
H A Dgpu_device.cc1582 std::vector<TfDeviceId> peer_gpu_ids; in CreateGPUDevice() local