Searched defs:dims_to_replicate (Results 1 – 2 of 2) sorted by relevance
1188 const HloSharding& sharding, absl::Span<const int64> dims_to_replicate) { in PartiallyReplicateTiledShardingOnDims()1238 std::vector<int64> dims_to_replicate(data_rank); in PartiallyReplicateTiledShardingOnAllDimsExcept() local
1156 std::vector<int64> dims_to_replicate; in GetShardingFromUser() local