From efebd64c95677b2cca39fdb0988a5cb1ed26aa29 Mon Sep 17 00:00:00 2001 From: Blake Hechtman Date: Tue, 20 Oct 2020 10:29:40 -0700 Subject: [PATCH] [XLA] Add select and scatter to pattern matchers. PiperOrigin-RevId: 338086463 Change-Id: I8ef6b68369fd5a454dbf8836dbd2085e0a9a7e56 --- 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 {