Migrate cos to structured kernel (#55564)
Summary: Pull Request resolved: https://github.com/pytorch/pytorch/pull/55564
Test Plan: Imported from OSS
Reviewed By: pbelevich
Differential Revision: D27653764
Pulled By: SplitInfinity
fbshipit-source-id: e13d07b2bc76d11e635de63a5d6d1a835da79e47
