Searched refs:SolveWithThomasAlgorithm (Results 1 – 1 of 1) sorted by relevance
150 SolveWithThomasAlgorithm(context, superdiag, diag, subdiag, rhs, x); in ComputeMatrix()396 void SolveWithThomasAlgorithm(OpKernelContext* context, in SolveWithThomasAlgorithm() function in tensorflow::TridiagonalSolveOp