code: def testTwoFills(self, is_graph, size):
      layout_x = Layout.batch_sharded(self.mesh, _MESH_DIM_X, rank=1)
      layout_y = Layout.batch_sharded(self.mesh, _MESH_DIM_Y, rank=1)
  
      def fn():
          return (
              array_ops.fill([size], 0.0, layout=layout_x),
              array_ops.fill([size], 0.0, layout=layout_y),
          )
  
      if is_graph:
          fn = polymorphic_function.function(fn)
  
      with api.default_mesh(self.mesh):
          dtensor_x, dtensor_y = fn()
      tensor = array_ops.zeros([size], layout=None)
  
      self.assertDTensorEqual(tensor, layout_x, dtensor_x)
      self.assertDTensorEqual(tensor, layout_y, dtensor_y)
  
file_path: tensorflow/tensorflow/dtensor/python/tests/array_ops_test.py
imports: import numpy
  from tensorflow.dtensor.python import api
  from tensorflow.dtensor.python import layout
  from tensorflow.dtensor.python.tests import test_util
  from tensorflow.python.eager.polymorphic_function import polymorphic_function
  from tensorflow.python.framework import combinations
  from tensorflow.python.ops import array_ops
  from tensorflow.python.platform import test
label_desc: Use image node_id 2 for calling the ArrayOpsTest obj's underlying member method code with example usage: obj.testTwoFills(is_graph, size) and returns: array_ops,  array_ops