[NNC] Add python bindings for Compute2 (#59350)
Summary: Pull Request resolved: https://github.com/pytorch/pytorch/pull/59350
Test Plan: Imported from OSS
Reviewed By: mruberry
Differential Revision: D28854806
Pulled By: huiguoo
fbshipit-source-id: b9091f9183249257aedc1eafb1992e0faf5dea82
