code: def _np_tridiagonal_solve(self, diags, rhs):
      return self.evaluate(
          tf.linalg.tridiagonal_solve(
              tf.constant(diags, dtype=tf.float32),
              tf.constant(rhs, dtype=tf.float32),
          )
      )
  
file_path: tf-quant-finance/tf_quant_finance/math/pde/steppers/douglas_adi_scheme_test.py
imports: import numpy
  import tensorflow.compat.v2
  import tf_quant_finance
  from tensorflow.python.framework import test_util
label_desc: Use image node_id 4 for calling the DouglasAdiSchemeTest obj's underlying member method code with example usage: obj._np_tridiagonal_solve(diags, rhs) and returns: self