Searched refs:tmp_tile_assignment_dimensions (Results 1 – 1 of 1) sorted by relevance
1923 auto tmp_tile_assignment_dimensions = in ReshardPartialReplicateWithAllToAll() local1925 tmp_tile_assignment_dimensions[to_replicate_dim] = 1; in ReshardPartialReplicateWithAllToAll()1926 tmp_tile_assignment_dimensions.push_back(num_replicas); in ReshardPartialReplicateWithAllToAll()1927 tmp_tile_assignment.Reshape(tmp_tile_assignment_dimensions); in ReshardPartialReplicateWithAllToAll()