aboutsummaryrefslogtreecommitdiffhomepage
path: root/tensorflow/compiler/xla/service/hlo_matchers.h
diff options
context:
space:
mode:
authorGravatar David Majnemer <majnemer@google.com>2018-08-28 17:08:01 -0700
committerGravatar TensorFlower Gardener <gardener@tensorflow.org>2018-08-28 17:12:30 -0700
commitde25baba51a7369bbdc5fd8051af0e9deb6a268d (patch)
tree4dd13806044b9e80128164ecaaf0d13ec2548173 /tensorflow/compiler/xla/service/hlo_matchers.h
parent4dcd00066fba2bd7c504c1bc35738f804de9df67 (diff)
[XLA] Add support for algebraic simplifications involving kIota
PiperOrigin-RevId: 210634966
Diffstat (limited to 'tensorflow/compiler/xla/service/hlo_matchers.h')
-rw-r--r--tensorflow/compiler/xla/service/hlo_matchers.h1
1 files changed, 1 insertions, 0 deletions
diff --git a/tensorflow/compiler/xla/service/hlo_matchers.h b/tensorflow/compiler/xla/service/hlo_matchers.h
index 9ace0d76e0..5502e565b6 100644
--- a/tensorflow/compiler/xla/service/hlo_matchers.h
+++ b/tensorflow/compiler/xla/service/hlo_matchers.h
@@ -188,6 +188,7 @@ HLO_MATCHER(Fusion);
HLO_MATCHER(Ge);
HLO_MATCHER(AfterAll);
HLO_MATCHER(Gt);
+HLO_MATCHER(Iota);
HLO_MATCHER(Infeed);
HLO_MATCHER(IsFinite);
HLO_MATCHER(Le);