Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/compiler/xla/service/
Dhlo_creation_utils.h137 const Shape& sort_shape, absl::Span<HloInstruction* const> operands,
Dhlo_creation_utils.cc289 const Shape& sort_shape, absl::Span<HloInstruction* const> operands, in MakeSortHlo() argument
308 sort_shape, dimension_to_sort, operands, compare_computation, is_stable)); in MakeSortHlo()