Searched refs:dims_to_replicate (Results 1 – 3 of 3) sorted by relevance
190 const HloSharding& sharding, absl::Span<const int64> dims_to_replicate);
1143 const HloSharding& sharding, absl::Span<const int64> dims_to_replicate) { in PartiallyReplicateTiledShardingOnDims() argument1148 for (int64 dim : dims_to_replicate) { in PartiallyReplicateTiledShardingOnDims()1164 return absl::c_linear_search(dims_to_replicate, a) < in PartiallyReplicateTiledShardingOnDims()1165 absl::c_linear_search(dims_to_replicate, b); in PartiallyReplicateTiledShardingOnDims()1172 for (int64 dim : dims_to_replicate) { in PartiallyReplicateTiledShardingOnDims()
1139 std::vector<int64> dims_to_replicate; in GetShardingFromUser() local1143 dims_to_replicate.push_back(i); in GetShardingFromUser()1156 user.sharding(), dims_to_replicate), in GetShardingFromUser()1157 dims_to_replicate); in GetShardingFromUser()