code: def _testTensorArrayWritePack(self, tf_dtype):
      with self.session(), self.test_scope():
          convert = _make_converter(tf_dtype)
  
          def fn():
              ta = tensor_array_ops.TensorArray(
                  dtype=tf_dtype, tensor_array_name="foo", size=3
              )
  
              w0 = ta.write(0, convert([[4.0, 5.0]]))
              w1 = w0.write(1, convert([[6.0, 7.0]]))
              w2 = w1.write(2, convert([[8.0, 9.0]]))
  
              return w2.stack()
  
          self.assertAllEqual(
              convert([[[4.0, 5.0]], [[6.0, 7.0]], [[8.0, 9.0]]]),
              self.evaluate(xla.compile(fn)[0]),
          )
  
imports: import numpy
  from tensorflow.compiler.tests import xla_test
  from tensorflow.python.compiler.xla import xla
  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 test_util
  from tensorflow.python.ops import array_ops
  from tensorflow.python.ops import control_flow_util
  from tensorflow.python.ops import gen_data_flow_ops
  from tensorflow.python.ops import gradients_impl
  from tensorflow.python.ops import math_ops
  from tensorflow.python.ops import resource_variable_ops
  from tensorflow.python.ops import tensor_array_grad
  from tensorflow.python.ops import tensor_array_ops
  from tensorflow.python.ops import variables
  from tensorflow.python.platform import test
label_desc: Use image node_id 2 for calling the TensorArrayTest obj's underlying member method code with example usage: obj._testTensorArrayWritePack(tf_dtype) and returns: w2