From 9e6096a772792c30b50cce658d3cefa63e84b2ee Mon Sep 17 00:00:00 2001 From: Andy Ly Date: Wed, 22 Apr 2020 15:51:58 -0700 Subject: [PATCH] Update TensorFlow MLIR ODS for tf.Maximum and tf.Minimum to support uint8. PiperOrigin-RevId: 307919461 Change-Id: I9ca18dacbd6da6e2d6c90aaa40f5593f39881198 --- .../compiler/mlir/tensorflow/ir/tf_generated_ops.td | 12 ++++++------ 1 file changed, 6 insertions(+), 6 deletions(-) diff --git a/tensorflow/compiler/mlir/tensorflow/ir/tf_generated_ops.td b/tensorflow/compiler/mlir/tensorflow/ir/tf_generated_ops.td index bd54689f994..8be7c69788b 100644 --- a/tensorflow/compiler/mlir/tensorflow/ir/tf_generated_ops.td +++ b/tensorflow/compiler/mlir/tensorflow/ir/tf_generated_ops.td @@ -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>;