code: def test_concat_in_function(self):
      @def_function.function
      def fn(a, b):
          values = tensor_array_ops.TensorArray(
              size=4,
              dtype=dtypes.string,
              element_shape=[None],
              infer_shape=False,
          )
          values = (
              (
                  values.write(0, a).write(
                      1, constant_op.constant([], dtypes.string)
                  )
              )
              .write(2, b)
              .write(3, constant_op.constant([], dtypes.string))
          )
          return values.concat()
  
      self.assertAllEqual(
          fn(["a", "b", "c"], ["c", "d", "e"]),
          [b"a", b"b", b"c", b"c", b"d", b"e"],
      )
  
file_path: tensorflow-upstream/tensorflow/python/ops/tensor_array_ops_test.py
imports: import numpy
  from tensorflow.python.eager import def_function
  from tensorflow.python.framework import constant_op
  from tensorflow.python.framework import dtypes
  from tensorflow.python.framework import test_util
  from tensorflow.python.ops import array_ops
  from tensorflow.python.ops import tensor_array_ops
  from tensorflow.python.platform import test
label_desc: Use image node_id 3 for calling the TensorArrayOpsTest obj's underlying member method code with example usage: obj.test_concat_in_function() and returns: values