SIGN IN SIGN UP

Add weights to model outputs (#4302)

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

Added a pass that flags weights with associated gradients and adds them to the models outputs. Updated the emitter to handle that something could be both a 'constant' and have an allocation_info

Reviewed By: tarun292

Differential Revision: D59929421

fbshipit-source-id: 6c3c85d260754b7c8271afc1f89877af68943fab
J
Jacob Szwejbka committed
301a017bb41557cbed7363bc0a16885eae75b8bb
Parent: 4ba11e3
Committed by Facebook GitHub Bot <facebook-github-bot@users.noreply.github.com> on 8/1/2024, 8:41:35 PM