class_docstr: Test for tri-diagonal matrix related ops.
code: def testGradientWithBatchDims(self):
      (
          y,
          diags,
          rhs,
          expected_grad_diags,
          expected_grad_rhs,
      ) = self._makeDataForGradientWithBatching()
  
      self._gradientTest(
          diags=diags,
          rhs=rhs,
          y=y,
          expected_grad_diags=expected_grad_diags,
          expected_grad_rhs=expected_grad_rhs,
      )
  
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 34 for calling the TridiagonalSolveOpsTest obj's underlying member method code with example usage: obj.testGradientWithBatchDims() without return types