pytorch

71f36909 - Revert "Revert "memory viz: Add colors for categories and a legend (#90587)"" (#96133)

Revert "Revert "memory viz: Add colors for categories and a legend (#90587)"" (#96133) This reverts commit b38b39c441f12be90fd5d7eafe74246d050665c8. Pull Request resolved: https://github.com/pytorch/pytorch/pull/96133 Approved by: https://github.com/bhosmer