tf.linalg.tridiagonal_matmul