Look in named-buffers of module for tensors (#47641)
Summary:
Pull Request resolved: https://github.com/pytorch/pytorch/pull/47641
ghstack-source-id: 116450114
Test Plan: Presubmit tests
Reviewed By: jamesr66a
Differential Revision: D24848318
fbshipit-source-id: f6ede3def9d6f1357c4fd3406f97721dea06b9f1
