Minor updates for upcast_tensor
Based on comments in https://github.com/pytorch/pytorch/pull/78459
Signed-off-by: Edward Z. Yang <ezyangfb.com>
Pull Request resolved: https://github.com/pytorch/pytorch/pull/78525
Approved by: https://github.com/albanD