Searched refs:valid_dims_to_replicate (Results 1 – 1 of 1) sorted by relevance
1265 std::vector<int64_t> valid_dims_to_replicate; in PartiallyReplicateTiledShardingOnDims() local1270 valid_dims_to_replicate.push_back(dim); in PartiallyReplicateTiledShardingOnDims()1282 return absl::c_linear_search(valid_dims_to_replicate, a) < in PartiallyReplicateTiledShardingOnDims()1283 absl::c_linear_search(valid_dims_to_replicate, b); in PartiallyReplicateTiledShardingOnDims()1290 for (int64_t dim : valid_dims_to_replicate) { in PartiallyReplicateTiledShardingOnDims()