code: def test_max_performance_type2(self):
      n = 2**10 * 4
      x = tf.range(n)
      x = tf.reshape(tf.random.shuffle(x), [2**10, 4])
  
      @tf.function
      def test_max2(input):
          y = tf.reduce_max(input, axis=1)
          return y
  
      Performance.time_log("1st run")
      test_max2(x)
      Performance.time_log("1st run")
  
      Performance.time_log("2nd run")
      test_max2(x)
      Performance.time_log("2nd run")
  
file_path: tf-encrypted/examples/benchmark/operation/plain_profile.py
imports: import sys
  import unittest
  import tensorflow
  from tf_encrypted.performance import Performance
label_desc: Use image node_id 4 for calling the TestPlainProfile obj's underlying member method code with example usage: obj.test_max_performance_type2() and returns: y