diff --git a/xla/service/pattern_matcher.h b/xla/service/pattern_matcher.h index eacd17ed0bfb4..f5267bc688e78 100644 --- a/xla/service/pattern_matcher.h +++ b/xla/service/pattern_matcher.h @@ -2816,6 +2816,7 @@ XLA_COMMUTATIVE_BINOP_PATTERN(Xor) .WithOperand(2, std::forward(arg2)); \ } XLA_TERNOP_PATTERN(Clamp); +XLA_TERNOP_PATTERN(RaggedDot); XLA_TERNOP_PATTERN(Select); XLA_TERNOP_PATTERN(SelectAndScatter); #undef XLA_TERNOP_PATTERN