Home
last modified time | relevance | path

Searched refs:compare_computation (Results 1 – 2 of 2) sorted by relevance

/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/service/
H A Dhlo_creation_utils.cc535 HloComputation* compare_computation; in MakeSortHlo() local
550 compare_computation = in MakeSortHlo()
553 sort_shape, dimension_to_sort, operands, compare_computation, is_stable)); in MakeSortHlo()
/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/service/spmd/
H A Dcustom_call_handler.cc196 auto compare_computation = in HandleCustomCallTopK() local
208 compare_computation, true)); in HandleCustomCallTopK()