There is not much of a benefit to reshape a from element vs reloading it. Updated to progagate shape manipulations into the output type of tensor.from_elements. Reviewed By: NatashaKnk Differential Revision: https://reviews.llvm.org/D118201
// ----