def _benchmark_np_matmul(self, m, transpose_b, num_iters):
    a = m.cpu().numpy()
    b = (a.T if transpose_b else a)
    func = (lambda : np.dot(a, b))
    self._run(func, num_iters)
def _benchmark_np_matmul(self, m, transpose_b, num_iters):
    a = m.cpu().numpy()
    b = (a.T if transpose_b else a)
    func = (lambda : np.dot(a, b))
    self._run(func, num_iters)