[distributed][Tensor Parallelism] Implement early transforms for column-wise and row-wise linear and embedding

[distributed][Tensor Parallelism] Implement early transforms for column-wise and row-wise linear and embedding #1861