Searched refs:NcclCliqueKey (Results 1 – 7 of 7) sorted by relevance
| /external/tensorflow/tensorflow/compiler/xla/service/gpu/ |
| D | gpu_executable_run_options.h | 33 class NcclCliqueKey {
35 explicit NcclCliqueKey(std::vector<GlobalDeviceId> devices);
38 friend H AbslHashValue(H h, const NcclCliqueKey& k) { in AbslHashValue()
41 friend bool operator==(const NcclCliqueKey& a, const NcclCliqueKey& b) {
54 std::function<StatusOr<std::string>(const NcclCliqueKey&)>;
|
| D | gpu_executable_run_options.cc | 23 NcclCliqueKey::NcclCliqueKey(std::vector<GlobalDeviceId> devices) in NcclCliqueKey() function in xla::gpu::NcclCliqueKey
26 std::string NcclCliqueKey::ToString() const { in ToString()
|
| D | nccl_utils.h | 132 const NcclCliqueKey& key,
134 const NcclCliqueKey&)>& value_factory) ABSL_LOCKS_EXCLUDED(mu_);
138 const std::function<void(const NcclCliqueKey&, const NcclClique&)>& fn)
143 absl::flat_hash_map<NcclCliqueKey, std::unique_ptr<NcclClique>> map_
|
| D | nccl_utils.cc | 151 const NcclCliqueKey& key, in CreateNcclClique()
233 key_, [&](const NcclCliqueKey& key) { in RunCollectiveOp()
248 NcclCliqueKey key_;
309 const NcclCliqueKey& key, in GetOrTryCreateIfAbsent()
311 const NcclCliqueKey&)>& value_factory) { in GetOrTryCreateIfAbsent()
322 const std::function<void(const NcclCliqueKey&, const NcclClique&)>& fn) { in ForEach() argument
|
| D | nccl_test_utils.cc | 27 NcclCliqueCache().ForEach([&](const NcclCliqueKey& k, const NcclClique&) { in DevicesWithOpenNcclChannels()
|
| /external/tensorflow/tensorflow/compiler/xla/pjrt/ |
| D | gpu_device.cc | 184 StatusOr<std::string> GetNcclUniqueId(const gpu::NcclCliqueKey& key);
192 absl::flat_hash_map<gpu::NcclCliqueKey, std::string> cache_
197 const gpu::NcclCliqueKey& key) { in GetNcclUniqueId()
299 [nccl_id_store](const gpu::NcclCliqueKey& key) { in BuildDistributedDevices()
|
| /external/tensorflow/tensorflow/compiler/xrt/kernels/ |
| D | xrt_execute_op.cc | 337 [&](const xla::gpu::NcclCliqueKey& key) -> xla::StatusOr<std::string> { in RunExecutable()
|