Add traces to specialize_autograd and lower_grad_of (2nd try)
Summary: Pull Request resolved: https://github.com/pytorch/pytorch/pull/22752
Differential Revision: D17543836
Pulled By: Krovatkin
fbshipit-source-id: 5cbca220943a580169bf60ac09780b6e67075d2b