Searched refs:compare_computation (Results 1 – 2 of 2) sorted by relevance
196 auto compare_computation = in HandleCustomCallTopK() local208 compare_computation, true)); in HandleCustomCallTopK()
535 HloComputation* compare_computation; in MakeSortHlo() local550 compare_computation = in MakeSortHlo()553 sort_shape, dimension_to_sort, operands, compare_computation, is_stable)); in MakeSortHlo()