pytorch
766f236b - Support params/buffers inside cond and map (#102310)

Commit
2 years ago
Support params/buffers inside cond and map (#102310) With #102022, params and buffers are always treated as special case of free variables. In this PR, I switch cond and map implementation to the this method and deprecate the old tracing mechanism. Pull Request resolved: https://github.com/pytorch/pytorch/pull/102310 Approved by: https://github.com/avikchaudhuri, https://github.com/zou3519
Committer
Parents
Loading