From bbaaccaab2e872645f91d3b0712263ad1d16c8c7 Mon Sep 17 00:00:00 2001
From: Renjie Liu <renjieliu@google.com>
Date: Sun, 19 Apr 2020 20:23:01 -0700
Subject: [PATCH] add SameOperandsAndResultsScale to prelu
PiperOrigin-RevId: 307326262
Change-Id: I1addc3aec63773cd92dd9739c30d5b5bf99b9327
---
tensorflow/compiler/mlir/lite/ir/tfl_ops.td | 4 +++-
1 file changed, 3 insertions(+), 1 deletion(-)
diff --git a/tensorflow/compiler/mlir/lite/ir/tfl_ops.td b/tensorflow/compiler/mlir/lite/ir/tfl_ops.td
index 4e3c1c0c5f4..314f6f2f3cc 100644
--- a/tensorflow/compiler/mlir/lite/ir/tfl_ops.td
+++ b/tensorflow/compiler/mlir/lite/ir/tfl_ops.td
@@ -2116,7 +2116,9 @@ def TFL_PowOp : TFL_Op<"pow", [ResultsBroadcastableShape,
let builders = [TFL_BroadcastableBinaryBuilder];
}
-def TFL_PReluOp : TFL_Op<"prelu", [NoSideEffect, TFL_GpuTargetOp]> {
+def TFL_PReluOp : TFL_Op<"prelu", [NoSideEffect,
+ TFL_GpuTargetOp,
+ SameOperandsAndResultsScale]> {
let summary = "Parameterized Relu operator";
let description = [{