Make aten_tensor_iterator ASAN safe (#37869)
Summary:
Pull Request resolved: https://github.com/pytorch/pytorch/pull/37869
Return type of `cpu_serial_kernel` functor should match type of the Tensor
Closes https://github.com/pytorch/pytorch/issues/37490
Test Plan: CI
Differential Revision: D21410450
fbshipit-source-id: 78081d7478fc8126cbd497625ba60ed17e253314
