move topk to use OpInfo (#55547)
Summary: Pull Request resolved: https://github.com/pytorch/pytorch/pull/55547
Test Plan: Imported from OSS
Reviewed By: VitalyFedyunin, mruberry
Differential Revision: D27649412
Pulled By: albanD
fbshipit-source-id: e36a5bb5703681b7f7647ca30d6f4a72faf5ed0e
