class_docstr: A Collection of KPI benchmarks.
class_docstr_tok: ["A","Collection","of","KPI","benchmarks","."]
code: def benchmark_tf_function_invocation_identity(self):
      x = tf.constant([[1.0, 2.0], [3.0, 4.0]])
  
      @tf.function
      def identity(x):
          return x
  
      def fn():
          with trace.Trace("tf.function-identity"):
              identity(x)
  
      self._run(fn, NUM_ITERATIONS)
  
file_path: tensorflow/tensorflow/python/eager/benchmarks/kpi_benchmark_test.py
imports: import gc
  import time
  import tensorflow
  from tensorflow.python.eager import benchmarks_test_base
  from tensorflow.python.eager import context
  from tensorflow.python.profiler import trace
label_desc: Use image node_id 6 for calling the KpiBenchmarks obj's underlying member method code with example usage: obj.benchmark_tf_function_invocation_identity() and returns: x