code: def testLogicalToPhysical(self, physical_shape, logical_shape):
      logical_to_physical = (
          mtf.simd_mesh_impl.auto_logical_to_physical_tpu(
              physical_shape, logical_shape
          )
      )
      self.assertCountEqual(
          logical_to_physical,
          list(range(mtf.list_product(physical_shape))),
      )
  
file_path: mesh/mesh_tensorflow/simd_mesh_impl_test.py
imports: from __future__ import absolute_import
  from __future__ import division
  from __future__ import print_function
  from absl.testing import parameterized
  import mesh_tensorflow
  import tensorflow.compat.v1
label_desc: Use image node_id 2 for calling the SimdMeshImplTest obj's underlying member method code with example usage: obj.testLogicalToPhysical(physical_shape, logical_shape) without return types