Searched refs:replica_groups_attr (Results 1 – 1 of 1) sorted by relevance
1038 auto replica_groups_attr = xla::HloFunctionImporter::ConvertReplicaGroups( in SetupCommonCollectiveOpAttributes() local1040 op->setAttr(replica_groups_attr.first, replica_groups_attr.second); in SetupCommonCollectiveOpAttributes()