all_to_one cuda support non-2d inputs

[IvanKobzarev]

Summary: Support non-2d tensors in all_to_one. Sometimes torchrec outputs are empty tensors.

This can be solved on python module side guarding empty tensors, but we d better not to have shape dependent logic for PT2 tracing.

Differential Revision: D57180617

[facebook-github-bot]

This pull request was exported from Phabricator. Differential Revision: D57180617

[netlify[bot]]

✅ Deploy Preview for pytorch-fbgemm-docs ready!

Name	Link
🔨 Latest commit	2e5438a4363c941d8e3fc13aa4afdbfd7317e68b
🔍 Latest deploy log	https://app.netlify.com/sites/pytorch-fbgemm-docs/deploys/663d419e7e36790008a55833
😎 Deploy Preview	https://deploy-preview-2575--pytorch-fbgemm-docs.netlify.app
📱 Preview on mobile	Toggle QR Code... Use your smartphone camera to open QR code link.

---

To edit notification comments on pull requests, go to your Netlify site configuration.

[facebook-github-bot]

This pull request was exported from Phabricator. Differential Revision: D57180617

[facebook-github-bot]

This pull request was exported from Phabricator. Differential Revision: D57180617

[facebook-github-bot]

This pull request was exported from Phabricator. Differential Revision: D57180617