code: def benchmarkMatrixExponentialOp(self):
      for shape in self.shapes:
          with ops.Graph().as_default(), session.Session(
              config=benchmark.benchmark_config()
          ) as sess, ops.device("/cpu:0"):
              matrix = self._GenerateMatrix(shape)
              expm = linalg_impl.matrix_exponential(matrix)
              self.evaluate(variables.global_variables_initializer())
              self.run_op_benchmark(
                  sess,
                  control_flow_ops.group(expm),
                  min_iters=25,
                  name="matrix_exponential_cpu_{shape}".format(
                      shape=shape
                  ),
              )
  
          if test.is_gpu_available(True):
              with ops.Graph().as_default(), session.Session(
                  config=benchmark.benchmark_config()
              ) as sess, ops.device("/gpu:0"):
                  matrix = self._GenerateMatrix(shape)
                  expm = linalg_impl.matrix_exponential(matrix)
                  self.evaluate(
                      variables.global_variables_initializer()
                  )
                  self.run_op_benchmark(
                      sess,
                      control_flow_ops.group(expm),
                      min_iters=25,
                      name="matrix_exponential_gpu_{shape}".format(
                          shape=shape
                      ),
                  )
  
imports: import itertools
  import numpy
  from tensorflow.python.client import session
  from tensorflow.python.framework import constant_op
  from tensorflow.python.framework import ops
  from tensorflow.python.framework import test_util
  from tensorflow.python.ops import array_ops
  from tensorflow.python.ops import control_flow_ops
  from tensorflow.python.ops import random_ops
  from tensorflow.python.ops import variables
  from tensorflow.python.ops.linalg import linalg_impl
  from tensorflow.python.platform import benchmark
  from tensorflow.python.platform import test
label_desc: Use image node_id 2 for calling the MatrixExponentialBenchmark obj's underlying member method code with example usage: obj.benchmarkMatrixExponentialOp() without return types