code: def _jit_tridiagonal_matmul(self, diagonals, rhs, diagonals_format):
      return linalg_impl.tridiagonal_matmul(
          diagonals, rhs, diagonals_format=diagonals_format
      )
  
imports: import numpy
  from tensorflow.compiler.tests import xla_test
  from tensorflow.python.eager import def_function
  from tensorflow.python.framework import constant_op
  from tensorflow.python.framework import dtypes
  from tensorflow.python.framework import ops
  from tensorflow.python.ops import array_ops
  from tensorflow.python.ops import array_ops_stack
  from tensorflow.python.ops import gradient_checker_v2
  from tensorflow.python.ops import math_ops
  from tensorflow.python.ops import stateless_random_ops
  from tensorflow.python.ops.linalg import linalg_impl
  from tensorflow.python.platform import googletest
label_desc: Use image node_id 1 for calling the TridiagonalMatMulOpsTest obj's underlying member method code with example usage: obj._jit_tridiagonal_matmul(diagonals, rhs, diagonals_format) and returns: linalg_impl