Searched refs:HandleTopK (Results 1 – 2 of 2) sorted by relevance
187 Status HandleTopK(HloInstruction* hlo);
2256 Status IrEmitter::HandleTopK(HloInstruction* hlo) { in HandleTopK() function in xla::cpu::IrEmitter2303 return HandleTopK(custom_call); in HandleCustomCall()