Searched refs:SolveForSizeOneOrTwo (Results 1 – 1 of 1) sorted by relevance
155 SolveForSizeOneOrTwo(context, diagonals.data(), rhs.data(), x.data(), m, in ComputeMatrix()223 void SolveForSizeOneOrTwo(OpKernelContext* context, const Scalar* diagonals, in SolveForSizeOneOrTwo() function in tensorflow::TridiagonalSolveOpGpuLinalg