code: def set_tensor_model_parallel_rank(rank):
      """Set tensor model parallel rank."""
      global _MPU_TENSOR_MODEL_PARALLEL_RANK
      _MPU_TENSOR_MODEL_PARALLEL_RANK = rank
  
imports: from typing import Tuple, Optional
  import warnings
  import os
  import torch
  from apex.transformer.log_util import get_transformer_logger
  from apex.transformer._ucc_util import HAS_UCC
label_desc: Use image node_id 31 for calling a global function with example usage: set_tensor_model_parallel_rank(rank) without return types