commit 5ce12db050dc78e1f5c968cd2713f63638ab7648
author Haoliang Zhang <haoliang@google.com> Mon Oct 07 15:37:54 2019 -0700
committer TensorFlower Gardener <gardener@tensorflow.org> Mon Oct 07 19:50:02 2019 -0700
tree 09d58223b434b02df59532fe8e6edb0c851d96e3
parent 2f6f7d9c6449d28e63625e37528c2ad234285abc

Correct op specification of tfl.gather. Indeed we support bool input.

PiperOrigin-RevId: 273392254

tensorflow/compiler/mlir/lite/ir/tfl_ops.td

diff --git a/tensorflow/compiler/mlir/lite/ir/tfl_ops.td b/tensorflow/compiler/mlir/lite/ir/tfl_ops.td
index d87d5cf..d5f811c 100644
--- a/tensorflow/compiler/mlir/lite/ir/tfl_ops.td
+++ b/tensorflow/compiler/mlir/lite/ir/tfl_ops.td
@@ -581,7 +581,7 @@
   }];
 
   let arguments = (ins
-    TensorOf<[F32, I8, I32, I64, TFL_Str, QI8, QUI8]>:$params,
+    TensorOf<[F32, I1, I8, I32, I64, TFL_Str, QI8, QUI8]>:$params,
     TensorOf<[I32, I64]>:$indices,
     I32Attr:$axis
   );
@@ -594,7 +594,7 @@
   ];
 
   let results = (outs
-    TensorOf<[F32, I16, I32, I64, TFL_Str, QI8, QUI8]>:$output
+    TensorOf<[F32, I1, I8, I32, I64, TFL_Str, QI8, QUI8]>:$output
   );
 
   let hasOptions = 1;