pytorch
a117fde8 - [Profiler] Apply TensorMetadata for Optimizer and nnModule (#86047)

Commit
3 years ago
[Profiler] Apply TensorMetadata for Optimizer and nnModule (#86047) Summary: - Use `TensorMetadat` struct in saving tensor info from Optimizer and nnModule. Test Plan: buck run mode/opt //caffe2/test:profiler Reviewed By: chaekit Differential Revision: D39682205 Pull Request resolved: https://github.com/pytorch/pytorch/pull/86047 Approved by: https://github.com/chaekit, https://github.com/robieta
Author
Committer
Parents
Loading