code: def test_compute_output_signature(
      self, batch_input_shape, batch_output_shape
  ):
      layer = tokenization_layers.WhitespaceTokenizer()
      self.assertEqual(
          layer.compute_output_signature(
              tf.TensorSpec(batch_input_shape, tf.string)
          ),
          tf.TensorSpec(batch_output_shape, tf.string),
      )
  
file_path: text/tensorflow_text/python/keras/layers/tokenization_layers_test.py
imports: from __future__ import absolute_import
  from __future__ import division
  from __future__ import print_function
  from absl.testing import parameterized
  import numpy
  import tensorflow
  from tensorflow_text.python.keras.layers import tokenization_layers
label_desc: Use image node_id 1 for calling the TokenizationLayersTest obj's underlying member method code with example usage: obj.test_compute_output_signature(batch_input_shape, batch_output_shape) without return types