class_docstr: Test for tri-diagonal matrix related ops.
code: def testWithTwoBatchingDimensions(self):
      self._test(
          diags=np.array(
              [
                  [_sample_diags, -_sample_diags, _sample_diags],
                  [-_sample_diags, _sample_diags, -_sample_diags],
              ]
          ),
          rhs=np.array(
              [
                  [_sample_rhs, 2 * _sample_rhs, 3 * _sample_rhs],
                  [4 * _sample_rhs, 5 * _sample_rhs, 6 * _sample_rhs],
              ]
          ),
          expected=np.array(
              [
                  [
                      _sample_result,
                      -2 * _sample_result,
                      3 * _sample_result,
                  ],
                  [
                      -4 * _sample_result,
                      5 * _sample_result,
                      -6 * _sample_result,
                  ],
              ]
          ),
      )
  
imports: import numpy
  from tensorflow.compiler.tests import xla_test
  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 test_util
  from tensorflow.python.ops import array_ops
  from tensorflow.python.ops import gradients
  from tensorflow.python.ops import math_ops
  from tensorflow.python.ops.linalg import linalg_impl
  from tensorflow.python.platform import test
label_desc: Use image node_id 18 for calling the TridiagonalSolveOpsTest obj's underlying member method code with example usage: obj.testWithTwoBatchingDimensions() without return types