[MLIR] Allow ShapedTypeComponents with attribute in inferReturnTensorTypes

Originally, inferReturnTensorTypes didn't support shaped type components
containing an attribute just because there wasn't any motivating use-case.
Removing that limitation and using it to set the encoding attribute for
RankedTensorType.

Updated the existing test to set result attribute based on the first operand,
if available.

Signed-off-by: Smit Hinsu <smittvhinsu@gmail.com>

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