[mlir] vector.multi_reduction canonicalizes to vector.shape_cast (or

vector.extract, if the result is a scalar) only if all reduction
dimensions are of size 1.

Differential Revision: https://reviews.llvm.org/D135333