Searched refs:temp_target_tile_dims (Results 1 – 1 of 1) sorted by relevance
1132 auto temp_target_tile_dims = sharding().tile_assignment().dimensions(); in ReshardWithAllToAll() local1133 temp_target_tile_dims[source_dim] = in ReshardWithAllToAll()1135 temp_target_tile_dims[target_dim] = in ReshardWithAllToAll()1137 temp_target_tile.Reshape(temp_target_tile_dims); in ReshardWithAllToAll()