code: def benchmarkCholeskyOp(self):
      for shape in self.shapes:
          with ops.Graph().as_default(), session.Session(
              config=benchmark.benchmark_config()
          ) as sess, ops.device("/cpu:0"):
              matrix = variables.Variable(self._GenerateMatrix(shape))
              l = linalg_ops.cholesky(matrix)
              self.evaluate(variables.global_variables_initializer())
              self.run_op_benchmark(
                  sess,
                  control_flow_ops.group(
                      l,
                  ),
                  min_iters=25,
                  name="cholesky_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("/device:GPU:0"):
                  matrix = variables.Variable(
                      self._GenerateMatrix(shape)
                  )
                  l = linalg_ops.cholesky(matrix)
                  self.evaluate(
                      variables.global_variables_initializer()
                  )
                  self.run_op_benchmark(
                      sess,
                      control_flow_ops.group(
                          l,
                      ),
                      min_iters=25,
                      name="cholesky_gpu_{shape}".format(shape=shape),
                  )
  
imports: import numpy
  from tensorflow.python.client import session
  from tensorflow.python.framework import constant_op
  from tensorflow.python.framework import dtypes
  from tensorflow.python.framework import errors_impl
  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 gradient_checker_v2
  from tensorflow.python.ops import linalg_ops
  from tensorflow.python.ops import math_ops
  from tensorflow.python.ops import stateless_random_ops
  from tensorflow.python.ops import variables
  from tensorflow.python.ops.linalg import linalg
  from tensorflow.python.platform import benchmark
  from tensorflow.python.platform import test
label_desc: Use image node_id 2 for calling the CholeskyBenchmark obj's underlying member method code with example usage: obj.benchmarkCholeskyOp() without return types