code: def testGreedySlowTPUVsNonTPU(self):
      decode_length = DECODE_LENGTH
  
      model, features = self._create_greedy_infer_model()
  
      with tf.variable_scope(tf.get_variable_scope(), reuse=True):
          slow_result_non_tpu = model._slow_greedy_infer(
              features, decode_length
          )["outputs"]
          slow_result_non_tpu = tf.squeeze(
              slow_result_non_tpu, axis=[2, 3]
          )
  
          slow_result_tpu = model._slow_greedy_infer_tpu(
              features, decode_length
          )["outputs"]
          slow_result_tpu = tf.squeeze(slow_result_tpu, axis=[2, 3])
  
      with self.test_session():
          slow_non_tpu_res = slow_result_non_tpu.eval()
          slow_tpu_res = slow_result_tpu.eval()
  
      self.assertEqual(
          slow_tpu_res.shape, (BATCH_SIZE, INPUT_LENGTH + decode_length)
      )
      self.assertAllClose(slow_tpu_res, slow_non_tpu_res)
  
imports: from __future__ import absolute_import
  from __future__ import division
  from __future__ import print_function
  import numpy
  from tensor2tensor.data_generators import problem_hparams
  from tensor2tensor.models import evolved_transformer
  from tensor2tensor.models import transformer
  import tensorflow.compat.v1
  from tensorflow.compat.v1 import estimator
label_desc: Use image node_id 6 for calling the EvolvedTransformerTest obj's underlying member method code with example usage: obj.testGreedySlowTPUVsNonTPU() without return types