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

Update TensorFlow MLIR ODS for tf.Maximum and tf.Minimum to support uint8.

Browse Source 
PiperOrigin-RevId: 307919461
Change-Id: I9ca18dacbd6da6e2d6c90aaa40f5593f39881198
This commit is contained in:
Andy Ly 2020-04-22 15:51:58 -07:00 committed by TensorFlower Gardener
parent 6d068076b9
commit 9e6096a772
1 changed files with 6 additions and 6 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

12
tensorflow/compiler/mlir/tensorflow/ir/tf_generated_ops.td
View File

@ -4714,12 +4714,12 @@ def TF_MaximumOp : TF_Op<"Maximum", [NoSideEffect, ResultsBroadcastableShape]>,
}];
let arguments = (ins
TensorOf<[BF16, F16, F32, F64, I16, I32, I64]>:$x,
TensorOf<[BF16, F16, F32, F64, I16, I32, I64]>:$y
TensorOf<[BF16, F16, F32, F64, I16, I32, I64, TF_Uint8]>:$x,
TensorOf<[BF16, F16, F32, F64, I16, I32, I64, TF_Uint8]>:$y
);
let results = (outs
TensorOf<[BF16, F16, F32, F64, I16, I32, I64]>:$z
TensorOf<[BF16, F16, F32, F64, I16, I32, I64, TF_Uint8]>:$z
);
TF_DerivedOperandTypeAttr T = TF_DerivedOperandTypeAttr<0>;
@ -4762,12 +4762,12 @@ def TF_MinimumOp : TF_Op<"Minimum", [NoSideEffect, ResultsBroadcastableShape]>,
}];
let arguments = (ins
TensorOf<[BF16, F16, F32, F64, I16, I32, I64]>:$x,
TensorOf<[BF16, F16, F32, F64, I16, I32, I64]>:$y
TensorOf<[BF16, F16, F32, F64, I16, I32, I64, TF_Uint8]>:$x,
TensorOf<[BF16, F16, F32, F64, I16, I32, I64, TF_Uint8]>:$y
);
let results = (outs
TensorOf<[BF16, F16, F32, F64, I16, I32, I64]>:$z
TensorOf<[BF16, F16, F32, F64, I16, I32, I64, TF_Uint8]>:$z
);
TF_DerivedOperandTypeAttr T = TF_DerivedOperandTypeAttr<0>;