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