code: def benchmarkSVDOp(self):
      for shape_ in self.shapes:
          with ops.Graph().as_default(), session.Session(
              config=benchmark.benchmark_config()
          ) as sess, ops.device("/cpu:0"):
              matrix_value = np.random.uniform(
                  low=-1.0, high=1.0, size=shape_
              ).astype(np.float32)
              matrix = variables.Variable(matrix_value)
              u, s, v = linalg_ops.svd(matrix)
              self.evaluate(variables.global_variables_initializer())
              self.run_op_benchmark(
                  sess,
                  control_flow_ops.group(u, s, v),
                  min_iters=25,
                  name="SVD_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_value = np.random.uniform(
                      low=-1.0, high=1.0, size=shape_
                  ).astype(np.float32)
                  matrix = variables.Variable(matrix_value)
                  u, s, v = linalg_ops.svd(matrix)
                  self.evaluate(
                      variables.global_variables_initializer()
                  )
                  self.run_op_benchmark(
                      sess,
                      control_flow_ops.group(u, s, v),
                      min_iters=25,
                      name="SVD_gpu_{shape}".format(shape=shape_),
                  )
  
imports: import numpy
  from tensorflow.python.client import session
  from tensorflow.python.eager import context
  from tensorflow.python.framework import constant_op
  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
  from tensorflow.python.ops import gradient_checker_v2
  from tensorflow.python.ops import gradients_impl
  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.platform import benchmark
  from tensorflow.python.platform import test
label_desc: Use image node_id 1 for calling the SVDBenchmark obj's underlying member method code with example usage: obj.benchmarkSVDOp() without return types