commit 68f84e3a8c78dbfa495163df0d6c1b600341d046
author Stella Laurenzo <laurenzo@google.com> Thu May 02 17:26:00 2019 -0700
committer Mehdi Amini <joker.eph@gmail.com> Mon May 06 08:25:57 2019 -0700
tree 7853f33e5eff85c032de191b48a63faf04416358
parent 800706fbdb725c3db0f7fc41e1f348bba0c6e5c3

    Add missing SameValueType trait to a couple of quantization ops.

--

PiperOrigin-RevId: 246422412

include/mlir/Quantization/QuantOps.td

diff --git a/include/mlir/Quantization/QuantOps.td b/include/mlir/Quantization/QuantOps.td
index 225f45ba..00415fa 100644
--- a/include/mlir/Quantization/QuantOps.td
+++ b/include/mlir/Quantization/QuantOps.td
@@ -104,7 +104,7 @@
 //===----------------------------------------------------------------------===//
 
 def quant_ConstFakeQuant : quant_Op<"const_fake_quant",
-                                    [NoSideEffect]> {
+                                    [SameValueType, NoSideEffect]> {
   let summary =
       "Simulates the effect of uniform quantization with const range.";
 
@@ -130,7 +130,7 @@
   );
 }
 
-def quant_StatisticsRefOp : quant_Op<"stats_ref", []> {
+def quant_StatisticsRefOp : quant_Op<"stats_ref", [SameValueType]> {
   let summary =
       "Indicates that statistics are resolved by reference.";
 
@@ -149,7 +149,7 @@
   let results = (outs quant_RealValueType);
 }
 
-def quant_StatisticsOp : quant_Op<"stats", []> {
+def quant_StatisticsOp : quant_Op<"stats", [SameValueType]> {
   let summary =
       "Identity op which associates statistics with the value.";
 
@@ -203,7 +203,7 @@
   }];
 }
 
-def quant_CoupledRefOp : quant_Op<"coupled_ref", []> {
+def quant_CoupledRefOp : quant_Op<"coupled_ref", [SameValueType]> {
   let summary =
       "Indicates that one point of the computation is coupled to another.";