Searched refs:selected_indices_sorted (Results 1 – 1 of 1) sorted by relevance
546 xla::XlaOp selected_indices_sorted = xla::GetTupleElement(output_tuple, 1); in Compile() local569 XlaGather(indices_sorted, scores_shape, selected_indices_sorted, in Compile()