Provide function interface for `remove_duplicate_output_args` (#65134)
authorKefei Lu <kefeilu@fb.com>
Thu, 16 Sep 2021 20:14:12 +0000 (13:14 -0700)
committerFacebook GitHub Bot <facebook-github-bot@users.noreply.github.com>
Thu, 16 Sep 2021 20:17:37 +0000 (13:17 -0700)
commit4c15f8e8b41c8f169d474b49136a67390a0bc60a
tree437d0ddad1158abb682d4f97996bf394ea64052e
parentf9c341fdf2a8b90017ad696b89cebbacf0c8fb95
Provide function interface for `remove_duplicate_output_args` (#65134)

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

So that its implementation can be abstracted and replaced

Test Plan: Run linter, CI

Reviewed By: 842974287

Differential Revision: D30966916

fbshipit-source-id: 92ec78c7410d0be14faecb0ba1eafdc74bab5a5d
torch/fx/experimental/fx2trt/passes/remove_duplicate_output_args.py