Revert back changes in test_custom_ops.cpp. (#55350)
Summary: Pull Request resolved: https://github.com/pytorch/pytorch/pull/55350
Test Plan: Imported from OSS
Reviewed By: albanD
Differential Revision: D27600413
Pulled By: ailzhang
fbshipit-source-id: 5e0d5f13fe3a51fcdccaad8af4d46cbe82795174
