[TensorBoard] fix #34954 (#36496)
Summary:
cc orionr sanekmelnikov
Pull Request resolved: https://github.com/pytorch/pytorch/pull/36496
Differential Revision: D21012775
Pulled By: natalialunova
fbshipit-source-id: 2dc978d70d457b511bd13a3399246ae0349ff8ca