[bazel] Add @pytorch in tools/bazel.bzl (#91424)
This is a follow-up from #89660
There is another place that needs to be updated.
I think this time I covered all of them...
Pull Request resolved: https://github.com/pytorch/pytorch/pull/91424
Approved by: https://github.com/malfet
