class_docstr: Benchmark matmul!
class_docstr_tok: ["Benchmark","matmul","!"]
code: def test_round(self, num_iters):
      dtypes = [np.float32, np.float64]
      for dtype in dtypes:
          for n, m, (transpose_a, transpose_b) in itertools.product(
              [512, 1024],
              [1, 8, 16, 128],
              [(False, False), (True, False), (False, True)],
          ):
              k = n
              self.run_test_gpu(
                  n, m, k, transpose_a, transpose_b, dtype, num_iters
              )
  
          for n, m, k, (transpose_a, transpose_b) in itertools.product(
              [200],
              [1, 8, 20],
              [10000],
              [(False, False), (True, False), (False, True)],
          ):
              self.run_test_gpu(
                  n, m, k, transpose_a, transpose_b, dtype, num_iters
              )
  
          for (n, m, k), (
              transpose_a,
              transpose_b,
          ) in itertools.product(
              [(200, 20, 20000), (1, 10000, 200)],
              [(False, False), (True, False), (False, True)],
          ):
              self.run_test_gpu(
                  n, m, k, transpose_a, transpose_b, dtype, num_iters
              )
  
file_path: tensorflow/tensorflow/python/ops/matmul_benchmark.py
imports: import itertools
  import time
  import numpy
  from tensorflow.python.client import session
  from tensorflow.python.framework import ops
  from tensorflow.python.ops import control_flow_ops
  from tensorflow.python.ops import math_ops
  from tensorflow.python.ops import random_ops
  from tensorflow.python.ops import variable_v1
  from tensorflow.python.ops import variables
  from tensorflow.python.platform import test
label_desc: Use image node_id 3 for calling the MatmulBenchmark obj's underlying member method code with example usage: obj.test_round(num_iters) without return types