Change all unary functions stubs to use TensorIteratorBase& (#56078)
Summary:
Pull Request resolved: https://github.com/pytorch/pytorch/pull/56078
This is in preparation for making all unary functions structured.
I don't actually have to make them structured yet as TensorIterator&
casts to TensorIteratorBase&
Signed-off-by: Edward Z. Yang <ezyang@fb.com>
Test Plan: Imported from OSS
Reviewed By: mruberry
Differential Revision: D27777768
Pulled By: ezyang
fbshipit-source-id: 05a3a95f200698eef72c5c74fff85fe881e1c4a3