From a92f7f998eea58f8fb9faef07dd3a8ba2de3f893 Mon Sep 17 00:00:00 2001 From: Blake Hechtman Date: Fri, 16 Oct 2020 06:56:11 -0700 Subject: [PATCH] [XLA] Add select and scatter to pattern mattcher PiperOrigin-RevId: 337497933 Change-Id: Ic8a2b1db182fa8d1d7b60650ad3773be346df9e7 --- tensorflow/compiler/xla/service/pattern_matcher.h | 1 + 1 file changed, 1 insertion(+) diff --git a/tensorflow/compiler/xla/service/pattern_matcher.h b/tensorflow/compiler/xla/service/pattern_matcher.h index eb29fa89098..8ebb522d6a8 100644 --- a/tensorflow/compiler/xla/service/pattern_matcher.h +++ b/tensorflow/compiler/xla/service/pattern_matcher.h @@ -2125,6 +2125,7 @@ XLA_BINOP_PATTERN(ShiftRightLogical) XLA_TERNOP_PATTERN(Clamp); XLA_TERNOP_PATTERN(Scatter); XLA_TERNOP_PATTERN(Select); +XLA_TERNOP_PATTERN(SelectAndScatter); #undef XLA_TERNOP_PATTERN namespace detail {