Searched defs:sort_shape (Results 1 – 2 of 2) sorted by relevance
198 const Shape sort_shape = ShapeUtil::MakeTupleShape( in HandleCustomCallTopK() local
419 const Shape& sort_shape, absl::Span<HloInstruction* const> operands, in MakeSortHlo()