code: def test_graph_mode(self):
      for tf_tokenizer in self.tf_tokenizers:
          compiled_tokenizer = tf.function(tf_tokenizer)
          for test_inputs in (
              self.test_sentences,
              self.paired_sentences,
          ):
              test_inputs = tf.constant(test_inputs)
              compiled_outputs = compiled_tokenizer(test_inputs)
              eager_outputs = tf_tokenizer(test_inputs)
  
              for key in eager_outputs.keys():
                  self.assertTrue(
                      tf.reduce_all(
                          eager_outputs[key] == compiled_outputs[key]
                      )
                  )
  
file_path: H2O/h2o_flexgen/benchmark/third_party/transformers/tests/models/bert/test_tokenization_bert_tf.py
imports: import unittest
  from pathlib import Path
  from tempfile import TemporaryDirectory
  from transformers import AutoConfig, TFAutoModel, is_tensorflow_text_available, is_tf_available
  from transformers.models.bert.tokenization_bert import BertTokenizer
  from transformers.testing_utils import require_tensorflow_text, slow
label_desc: Use image node_id 4 for calling the BertTokenizationTest obj's underlying member method code with example usage: obj.test_graph_mode() without return types