code: def calc_expected_tensor_model_paralell_rank(
      rank: int,
      tensor_model_parallel_world_size: int,
  ) -> int:
      return rank % tensor_model_parallel_world_size
  
file_path: apex/tests/L0/run_transformer/test_parallel_state.py
imports: import logging
  import os
  from torch.testing._internal import common_utils
  from apex.transformer import parallel_state
  from apex.transformer.testing.distributed_test_base import NcclDistributedTestBase
  from apex.transformer.testing.distributed_test_base import UccDistributedTestBase
label_desc: Use image node_id 1 for calling a global function with example usage: calc_expected_tensor_model_paralell_rank(rank, tensor_model_parallel_world_size) and returns: unknown