Searched defs:HloSortInstruction (Results 1 – 3 of 3) sorted by relevance
80 TopkRewriter rewriter([](const HloSortInstruction*, int64) { return true; }); in TEST_F()110 TopkRewriter rewriter([](const HloSortInstruction*, int64) { return true; }); in TEST_F()140 TopkRewriter rewriter([](const HloSortInstruction*, int64) { return true; }); in TEST_F()
28 explicit TopkRewriter(std::function<bool(const HloSortInstruction*, int64)> in TopkRewriter()
945 HloSortInstruction::HloSortInstruction( in HloSortInstruction() function in xla::HloSortInstruction