Affine exprs compose better than arith ops. Differential Revision: https://reviews.llvm.org/D132456
tensor.cast
tensor.collapse_shape