Calculate Tensor Parallel Communication Cost | TensorTonic