experiments/STT-tensorflow
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

add SameOperandsAndResultsScale to prelu

Browse Source 
PiperOrigin-RevId: 307326262
Change-Id: I1addc3aec63773cd92dd9739c30d5b5bf99b9327
This commit is contained in:
Renjie Liu 2020-04-19 20:23:01 -07:00 committed by TensorFlower Gardener
parent 62a6012a0c
commit bbaaccaab2
1 changed files with 3 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
tensorflow/compiler/mlir/lite/ir/tfl_ops.td
View File

@ -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 = [{