Searched refs:HloInstructionPredicate (Results 1 – 1 of 1) sorted by relevance
277 using HloInstructionPredicate = std::function<bool(const HloInstruction*)>; variable281 HloPrintOptions& set_print_instruction(HloInstructionPredicate callback) { in set_print_instruction()360 HloInstructionPredicate print_instruction_ = [](const HloInstruction* instr) {