Searched refs:sort_result (Results 1 – 2 of 2) sorted by relevance
/external/tensorflow/tensorflow/compiler/xla/client/lib/ |
D | sorting.cc | 58 XlaOp sort_result = in TopK() local 68 XlaOp values = Slice(GetTupleElement(sort_result, 0), start_indices, in TopK() 70 XlaOp indices = Slice(GetTupleElement(sort_result, 1), start_indices, in TopK() 120 XlaOp sort_result = in TopKWithPartitions() local 132 values = Slice(GetTupleElement(sort_result, 0), start_indices, in TopKWithPartitions() 134 indices = Slice(GetTupleElement(sort_result, 1), start_indices, in TopKWithPartitions() 151 XlaOp sort_result = in TopKWithPartitions() local 160 XlaOp values = Slice(GetTupleElement(sort_result, 0), start_indices, in TopKWithPartitions() 162 XlaOp indices = Slice(GetTupleElement(sort_result, 1), start_indices, in TopKWithPartitions()
|
D | self_adjoint_eig.cc | 351 XlaOp sort_result = in SortByEigenvalues() local 356 result.w = GetMatrixDiagonal(GetTupleElement(sort_result, 0)); in SortByEigenvalues() 357 result.v = GetTupleElement(sort_result, 1); in SortByEigenvalues()
|