Fix tensor_parallel function calls to megatron-lm
# What does this PR do ?

Megatron-LM has changed its function APIs in tensor_parallel, where a lot of params (e.g., `use_cpu_initialization`) are moved into a config object (e.g., `megatron.core.model_parallel_config.ModelParallelConfig`), for example [here](https://github.com/NVIDIA/Megatron-LM/blob/main/megatron/core/tensor_parallel/layers.py#L148)

This PR fixes such errors when trying to use those functions with the latest Megatron-LM codebase, but there may be some other missed functions. 

**Collection**: [NLP]

