code: def _GetLargeMatrixSolveLsOpTests(
      dtype, use_placeholder, fast, l2_regularizer
  ):
      def LargeBatchSquare(self):
          np.random.seed(1)
          num_rhs = 1
          matrix_shape = (127, 127)
          matrix = (
              np.random.uniform(
                  low=-1.0, high=1.0, size=np.prod(matrix_shape)
              )
              .reshape(matrix_shape)
              .astype(np.float32)
          )
          rhs = np.ones([matrix_shape[0], num_rhs]).astype(np.float32)
          self._verifySolve(
              matrix,
              rhs,
              dtype,
              use_placeholder,
              fast,
              l2_regularizer,
              batch_shape=(16, 8),
          )
  
      def LargeBatchOverdetermined(self):
          np.random.seed(1)
          num_rhs = 1
          matrix_shape = (127, 64)
          matrix = (
              np.random.uniform(
                  low=-1.0, high=1.0, size=np.prod(matrix_shape)
              )
              .reshape(matrix_shape)
              .astype(np.float32)
          )
          rhs = np.ones([matrix_shape[0], num_rhs]).astype(np.float32)
          self._verifySolve(
              matrix,
              rhs,
              dtype,
              use_placeholder,
              fast,
              l2_regularizer,
              batch_shape=(16, 8),
          )
  
      def LargeBatchUnderdetermined(self):
          np.random.seed(1)
          num_rhs = 1
          matrix_shape = (64, 127)
          matrix = (
              np.random.uniform(
                  low=-1.0, high=1.0, size=np.prod(matrix_shape)
              )
              .reshape(matrix_shape)
              .astype(np.float32)
          )
          rhs = np.ones([matrix_shape[0], num_rhs]).astype(np.float32)
          self._verifySolve(
              matrix,
              rhs,
              dtype,
              use_placeholder,
              fast,
              l2_regularizer,
              batch_shape=(16, 8),
          )
  
      return (
          LargeBatchSquare,
          LargeBatchOverdetermined,
          LargeBatchUnderdetermined,
      )
  
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 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 linalg_ops
  from tensorflow.python.ops import math_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 5 for calling a global function with example usage: _GetLargeMatrixSolveLsOpTests(dtype, use_placeholder, fast, l2_regularizer) and returns: LargeBatchSquare,  LargeBatchOverdetermined,  LargeBatchUnderdetermined