code: def test_tensor_input_2d(self):
      token_begin_offsets = tf.constant(
          [[0, 4, 8, 12, 17], [0, 4, 8, 12, 17]]
      )
      token_end_offsets = tf.constant(
          [[3, 7, 11, 16, 20], [3, 7, 11, 16, 19]]
      )
      boise_tags = tf.constant(
          [
              ["O", "B-animal", "I-animal", "E-animal", "O"],
              ["O", "O", "O", "S-loc", "O"],
          ]
      )
  
      expected_span_begin_offsets = tf.ragged.constant([[4], [12]])
      expected_span_end_offsets = tf.ragged.constant([[16], [16]])
      expected_span_type = tf.ragged.constant([["animal"], ["loc"]])
      (
          span_begin_offsets,
          span_end_offsets,
          span_type,
      ) = tf_text.boise_tags_to_offsets(
          token_begin_offsets, token_end_offsets, boise_tags
      )
  
      self.assertAllEqual(
          span_begin_offsets, expected_span_begin_offsets
      )
      self.assertAllEqual(span_end_offsets, expected_span_end_offsets)
      self.assertAllEqual(span_type, expected_span_type)
  
imports: from __future__ import absolute_import
  from __future__ import division
  from __future__ import print_function
  import tensorflow
  import tensorflow_text
  from tensorflow.python.framework import test_util
  from tensorflow.python.platform import test
label_desc: Use image node_id 6 for calling the BoiseTagsToOffsetsTest obj's underlying member method code with example usage: obj.test_tensor_input_2d() without return types