[PyTorch] Update Bazel build for TensorPipe (#54416)
Summary:
Pull Request resolved: https://github.com/pytorch/pytorch/pull/54416
Once D27230990 lands, we'll need this for TensorPipe to be built with Bazel.
ghstack-source-id: 124512701
Test Plan: None for now.
Reviewed By: beauby
Differential Revision: D27231000
fbshipit-source-id: 474cc1b23118703ecb47ed4b8e0c5b000572eae8
