Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/compiler/xla/pjrt/
Dgpu_device.cc296 auto nccl_id_store = std::make_shared<NcclIdStore>( in BuildDistributedDevices() local
299 [nccl_id_store](const gpu::NcclCliqueKey& key) { in BuildDistributedDevices()
300 return nccl_id_store->GetNcclUniqueId(key); in BuildDistributedDevices()