Searched defs:NcclCollectivePermuteConfig (Results 1 – 1 of 1) sorted by relevance
31 struct NcclCollectivePermuteConfig : public NcclCollectiveConfig { struct37 struct SourceTargetMapEntry {42 absl::flat_hash_map<int64, SourceTargetMapEntry> id_to_source_target;49 SourceTargetMapEntry GetSourceTarget(int64_t id) const { in GetSourceTarget()