Searched refs:match_opcode (Results 1 – 2 of 2) sorted by relevance
27 auto match_opcode = [&](HloOpcode opcode) { in MatchReductionComputation() local38 if (match_opcode(HloOpcode::kAdd)) { in MatchReductionComputation()40 } else if (match_opcode(HloOpcode::kMultiply)) { in MatchReductionComputation()42 } else if (match_opcode(HloOpcode::kMinimum) || in MatchReductionComputation()43 (type == PRED && match_opcode(HloOpcode::kAnd))) { in MatchReductionComputation()45 } else if (match_opcode(HloOpcode::kMaximum) || in MatchReductionComputation()46 (type == PRED && match_opcode(HloOpcode::kOr))) { in MatchReductionComputation()
100 string match_opcode = expect_hlo ? "%[^=]+= [^ ]+ ([^(]+)\\(.*" local105 match_bytes_per_cycle, separator, match_opcode);