Update parent module attributes when sharding with TP (#44421)
* feat: update parent module attribute when sharding weights for TP
* style: remove added space
* test: add unit tests for TensorParallelLayer
* test: add tests/tensor_parallel/test_tensor_parallel.py to CI
* test: add mark to enable test discovery
---------
Co-authored-by: Ferdinand Mom <47445085+3outeille@users.noreply.github.com>