code: def testMapDefunCancelledCorrectly(self):
      @def_function.function(
          input_signature=[tensor_spec.TensorSpec([5], dtypes.int64)]
      )
      def defun(x):
          # x has leading dimension 5, this will raise an error
          return array_ops.gather(x, 10)
  
      c = array_ops.tile(
          array_ops.expand_dims(
              constant_op.constant([1, 2, 3, 4, 5], dtype=dtypes.int64),
              0,
          ),
          [100, 1],
      )
      map_defun_op = map_defun.map_defun(
          defun, [c], [dtypes.int64], [()]
      )[0]
      with self.assertRaisesRegex(
          errors.InvalidArgumentError,
          r"indices = 10 is not in \[0, 5\)",
      ):
          self.evaluate(map_defun_op)
  
imports: import time
  from absl.testing import parameterized
  from tensorflow.python.client import session
  from tensorflow.python.data.experimental.ops import map_defun
  from tensorflow.python.data.kernel_tests import test_base
  from tensorflow.python.eager import def_function
  from tensorflow.python.framework import combinations
  from tensorflow.python.framework import constant_op
  from tensorflow.python.framework import dtypes
  from tensorflow.python.framework import errors
  from tensorflow.python.framework import ops
  from tensorflow.python.framework import sparse_tensor
  from tensorflow.python.framework import tensor_spec
  from tensorflow.python.ops import array_ops
  from tensorflow.python.ops import array_ops_stack
  from tensorflow.python.ops import check_ops
  from tensorflow.python.ops import data_flow_ops
  from tensorflow.python.ops import math_ops
  from tensorflow.python.ops import sparse_ops
  from tensorflow.python.platform import test
label_desc: Use image node_id 10 for calling the MapDefunTest obj's underlying member method code with example usage: obj.testMapDefunCancelledCorrectly() and returns: array_ops