Searched refs:tmp_full_shape (Results 1 – 1 of 1) sorted by relevance
2100 auto tmp_full_shape = tmp_shard_shape; in HandleReshape() local2101 tmp_full_shape.set_dimensions( in HandleReshape()2105 PartitionedHlo(tmp_reshape, tmp_full_shape, MakePartitioningState()); in HandleReshape()