From 7e54bf3113361e36f1a49e71a6ecbcd10ddf7015 Mon Sep 17 00:00:00 2001 From: Blake Hechtman Date: Wed, 14 Oct 2020 10:45:15 -0700 Subject: [PATCH] [XLA] Add pattern matcher for select and scatter. PiperOrigin-RevId: 337122945 Change-Id: I9aaa3cdcb3cb3379492e4697ada514124995a8ce --- 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 {