code: def setUp(self):
      super(OperationSplittabilityTest, self).setUp()
      self.graph = mtf.Graph()
      self.mesh = mtf.Mesh(self.graph, "my_mesh")
  
      self.a_dim = mtf.Dimension("a", 5)
      self.b_dim = mtf.Dimension("b", 10)
      self.c_dim = mtf.Dimension("c", 15)
  
      self.ab_shape = mtf.Shape([self.a_dim, self.b_dim])
      self.x = mtf.zeros(self.mesh, self.ab_shape)
  
      self.batch_dim = mtf.Dimension("batch", 100)
      self.grid_h_dim = mtf.Dimension("grid_h", 10)
      self.grid_w_dim = mtf.Dimension("grid_w", 10)
      self.filter_h_dim = mtf.Dimension("filter_h", 5)
      self.filter_w_dim = mtf.Dimension("filter_w", 5)
      self.in_dim = mtf.Dimension("in", 10)
      self.out_dim = mtf.Dimension("out", 10)
      self.image = mtf.zeros(
          self.mesh,
          [
              self.batch_dim,
              self.grid_h_dim,
              self.grid_w_dim,
              self.in_dim,
          ],
      )
  
imports: from __future__ import absolute_import
  from __future__ import division
  from __future__ import print_function
  from absl.testing import parameterized
  import mesh_tensorflow
  from mesh_tensorflow import test_utils
  import numpy
  import tensorflow.compat.v1
  from tensorflow.python.framework import test_util
label_desc: Use image node_id 1 for calling the OperationSplittabilityTest obj's underlying member method code with example usage: obj.setUp() without return types