commit b806eee56bf7ae175249f23f00fa49d91d060a46
author Stephan Herhut <herhut@google.com> Tue Dec 08 01:02:59 2020 -0800
committer TensorFlower Gardener <gardener@tensorflow.org> Tue Dec 08 01:25:26 2020 -0800
tree daff65ee8b8181fecae51c2c73c232673f1b369f
parent d2ced738aacaa2fac5b3dc4fa16fdd8f87fa5e54

Allow legalization of binary unranked and/or operations from tf to chlo.

PiperOrigin-RevId: 346270805
Change-Id: Ifc5facb1393633bc2a84f267f8a928ee146afc4a

tensorflow/compiler/mlir/xla/transforms/legalize_tf_patterns.td

diff --git a/tensorflow/compiler/mlir/xla/transforms/legalize_tf_patterns.td b/tensorflow/compiler/mlir/xla/transforms/legalize_tf_patterns.td
index b363e2a..1010c31 100644
--- a/tensorflow/compiler/mlir/xla/transforms/legalize_tf_patterns.td
+++ b/tensorflow/compiler/mlir/xla/transforms/legalize_tf_patterns.td
@@ -193,7 +193,7 @@
 //===----------------------------------------------------------------------===//
 
 class DirectLogicalBinaryPat<Op FromOp, Op ToOp>
-  : Pat<(FromOp AnyRankedTensor:$l, AnyRankedTensor:$r),
+  : Pat<(FromOp AnyTensor:$l, AnyTensor:$r),
         (ToOp $l, $r, (BinBroadcastDimensions $l, $r)),
         [(SignedIntTensor $l)]>;