Fix `aten::squeeze.dims` shape function (#98078)
Signed-Off By: Vivek Khandelwal <vivek@nod-labs.com>
Fixes https://github.com/llvm/torch-mlir/issues/1690#issuecomment-1491931180.
Pull Request resolved: https://github.com/pytorch/pytorch/pull/98078
Approved by: https://github.com/davidberard98
