class_docstr: Test for tri-diagonal matrix related ops.
code: def testGradientWithMultipleRhs(self):
      self._gradientTest(
          diags=_sample_diags,
          rhs=[[1, 2], [2, 4], [3, 6], [4, 8]],
          y=[[1, 5], [2, 6], [3, 7], [4, 8]],
          expected_grad_diags=[
              [-20, 28, -60, 0],
              [36, -35, 60, 80],
              [0, 63, -75, -80],
          ],
          expected_grad_rhs=[[0, 2], [1, 3], [1, 7], [0, -10]],
      )
  
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 32 for calling the TridiagonalSolveOpsTest obj's underlying member method code with example usage: obj.testGradientWithMultipleRhs() without return types