Run function check and out check in TestTensorDeviceOps (#43830)
Summary: Pull Request resolved: https://github.com/pytorch/pytorch/pull/43830
Reviewed By: ailzhang
Differential Revision: D23438101
Pulled By: mruberry
fbshipit-source-id: b581ce779ea2f50ea8dfec51d5469031ec7a0a67