code: def _create_mesh(self, mesh_dims, topology):
      global_ids = test_util.create_device_ids_array(topology)
      local_ids = np.ravel(global_ids).tolist()
      mesh_dict = {}
      for device in ("CPU", "GPU", "TPU"):
          mesh_dict[device] = Mesh(
              mesh_dims,
              global_ids,
              local_ids,
              test_util.create_device_list(topology, device),
          )
  
      return self.configTestMesh(mesh_dict)
  
imports: from absl.testing import parameterized
  import numpy
  from tensorflow.dtensor.python import layout
  from tensorflow.dtensor.python import numpy_util
  from tensorflow.dtensor.python.tests import test_util
  from tensorflow.python.eager import backprop
  from tensorflow.python.eager.polymorphic_function import polymorphic_function
  from tensorflow.python.framework import constant_op
  from tensorflow.python.framework import dtypes
  from tensorflow.python.ops import nn_ops
  from tensorflow.python.ops import special_math_ops
  from tensorflow.python.platform import test
label_desc: Use image node_id 2 for calling the SpatiallyPartitionedConvOpTest obj's underlying member method code with example usage: obj._create_mesh(mesh_dims, topology) and returns: self