Searched defs:HloSortInstruction (Results 1 – 3 of 3) sorted by relevance
134 [](const HloSortInstruction*, int64_t) { return true; }); in TEST_F()165 [](const HloSortInstruction*, int64_t) { return true; }); in TEST_F()196 [](const HloSortInstruction*, int64_t) { return true; }); in TEST_F()227 [](const HloSortInstruction*, int64_t) { return true; }); in TEST_F()258 [](const HloSortInstruction*, int64_t) { return true; }); in TEST_F()
30 explicit TopkRewriter(std::function<bool(const HloSortInstruction*, int64_t)> in TopkRewriter()
1178 HloSortInstruction::HloSortInstruction( in HloSortInstruction() function in xla::HloSortInstruction