Searched refs:sort_shape (Results 1 – 2 of 2) sorted by relevance
137 const Shape& sort_shape, absl::Span<HloInstruction* const> operands,
289 const Shape& sort_shape, absl::Span<HloInstruction* const> operands, in MakeSortHlo() argument308 sort_shape, dimension_to_sort, operands, compare_computation, is_stable)); in MakeSortHlo()