Searched refs:ConvertReplicaGroups (Results 1 – 7 of 7) sorted by relevance
35 StatusOr<std::vector<ReplicaGroup>> ConvertReplicaGroups(
76 static mlir::NamedAttribute ConvertReplicaGroups(
90 StatusOr<std::vector<ReplicaGroup>> ConvertReplicaGroups( in ConvertReplicaGroups() function
557 ConvertReplicaGroups(all_reduce->replica_groups(), builder_)); in ImportInstructionImpl()952 mlir::NamedAttribute HloFunctionImporter::ConvertReplicaGroups( in ConvertReplicaGroups() function in xla::HloFunctionImporter
178 return xla::ConvertReplicaGroups(groups).ValueOrDie(); in Convert_replica_groups()
89 ConvertReplicaGroups(op.replica_groups()).ValueOrDie(); in GetNcclCollectiveConfigForMlir()
1038 auto replica_groups_attr = xla::HloFunctionImporter::ConvertReplicaGroups( in SetupCommonCollectiveOpAttributes()