[acc_ops] Add support for torch variants of squeeze and mul (#65037)
authorJordan Fix <jfix@fb.com>
Thu, 16 Sep 2021 02:39:41 +0000 (19:39 -0700)
committerFacebook GitHub Bot <facebook-github-bot@users.noreply.github.com>
Thu, 16 Sep 2021 02:41:04 +0000 (19:41 -0700)
commit2bb898e039d90d3897cc0fbd1886f2b2fe4dbcfb
tree03520d4fc0a7480937c45f631708a82c1e014d7b
parent206646d6ed692de8aee663dda7e79e8fc5f42f73
[acc_ops] Add support for torch variants of squeeze and mul (#65037)

Summary:
Pull Request resolved: https://github.com/pytorch/pytorch/pull/65037

att

Test Plan: updated unit tests

Reviewed By: yuhc

Differential Revision: D30952224

fbshipit-source-id: aaf75b27b4fc6c0436ba7bfcf324f761b900171b
torch/fx/experimental/fx_acc/acc_ops.py