Add communication dtypes for all-gathers and reduce scatters in depth tensor parallelism

Add communication dtypes for all-gathers and reduce scatters in depth tensor parallelism #119