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()
|