Searched refs:sort_result (Results 1 – 2 of 2) sorted by relevance
36 XlaOp sort_result = in TopK() local46 XlaOp values = Neg(Slice(GetTupleElement(sort_result, 0), start_indices, in TopK()48 XlaOp indices = Slice(GetTupleElement(sort_result, 1), start_indices, in TopK()
351 XlaOp sort_result = in SortByEigenvalues() local356 result.w = GetMatrixDiagonal(GetTupleElement(sort_result, 0)); in SortByEigenvalues()357 result.v = GetTupleElement(sort_result, 1); in SortByEigenvalues()