Searched defs:subgroup_types (Results 1 – 7 of 7) sorted by relevance
236 std::vector<OpSharding::Type> subgroup_types = {OpSharding::MANUAL}; in TEST() local256 std::vector<OpSharding::Type> subgroup_types = {OpSharding::REPLICATED, in TEST() local277 std::vector<OpSharding::Type> subgroup_types = {OpSharding::MANUAL, in TEST() local
293 const std::vector<OpSharding::Type>& subgroup_types() const { in subgroup_types() function
245 std::vector<OpSharding::Type> subgroup_types; in MergeShardingIfCompatible() local342 std::vector<OpSharding::Type> subgroup_types; in TransposeSharding() local469 auto subgroup_types = sharding.subgroup_types(); in ReshapeSharding() local1301 std::vector<OpSharding::Type> subgroup_types; in PartiallyReplicateTiledShardingOnDims() local1732 std::vector<OpSharding::Type> subgroup_types; in UngroupSharding() local
126 absl::Span<const OpSharding::Type> subgroup_types, in Subgroup()649 std::vector<OpSharding::Type> subgroup_types; in FromProto() local
983 std::vector<OpSharding::Type> subgroup_types; in RefineManualAutoShardingFromAuto() local
3238 std::vector<OpSharding::Type> subgroup_types = {OpSharding::MANUAL, local
3051 std::vector<OpSharding::Type> subgroup_types; in ParseSingleSharding() local