SIGN IN SIGN UP

find bias for mm in all nodes (#1730)

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

Previously we had the following todo
```
# TODO also find bias in input args (not just in params)
```

This is because we searched for bias within the params of the source partition. However with lifted graphs, this may not always be the case. Lifted params may not be in params but will generaly be found in all nodes. We change the logic here to search for the bias node within all the nodes. We do this by just finding any param node which is not the weight node within the source partition. This must be the bias node.

Reviewed By: tarun292, digantdesai

Differential Revision: D53143457

fbshipit-source-id: 54f199d9e65d26c592b8dfd9aca4518706c8d89c
M
Max Ren committed
e77a56b91f7c024d4ac8b06821516c52caad7f99
Parent: 6a1b70a
Committed by Facebook GitHub Bot <facebook-github-bot@users.noreply.github.com> on 1/29/2024, 7:08:00 PM