From 9877c212c8d2a6236adaa85d0fe3a2c35819dfed Mon Sep 17 00:00:00 2001 From: Smit Hinsu Date: Fri, 20 Sep 2019 15:37:42 -0700 Subject: [PATCH] Add SameOperandsAndResultType trait to SignOp and ReverseOp in HLO Currently, these ops are using SameOperandsAndResultShape and SameOperandsAndResultElementType, respectively. PiperOrigin-RevId: 270364879 --- tensorflow/compiler/mlir/xla/ir/hlo_ops.td | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/tensorflow/compiler/mlir/xla/ir/hlo_ops.td b/tensorflow/compiler/mlir/xla/ir/hlo_ops.td index 79f2d79ae23..342e763e367 100644 --- a/tensorflow/compiler/mlir/xla/ir/hlo_ops.td +++ b/tensorflow/compiler/mlir/xla/ir/hlo_ops.td @@ -135,7 +135,7 @@ def HLO_NegOp: HLO_UnaryElementwiseOp<"neg", [NoSideEffect, SameOperandsAndResul def HLO_RsqrtOp: HLO_UnaryElementwiseOp<"rsqrt", [NoSideEffect, SameOperandsAndResultType]>, BASE_HLO_RsqrtOp; -def HLO_SignOp: HLO_UnaryElementwiseOp<"sign", [NoSideEffect, SameOperandsAndResultShape]>, BASE_HLO_SignOp; +def HLO_SignOp: HLO_UnaryElementwiseOp<"sign", [NoSideEffect, SameOperandsAndResultType]>, BASE_HLO_SignOp; def HLO_TanhOp: HLO_UnaryElementwiseOp<"tanh", [ResultsAreFloatLike, NoSideEffect, SameOperandsAndResultType]>, BASE_HLO_TanhOp; @@ -481,7 +481,7 @@ def HLO_SelectOp: HLO_Op<"select", [NoSideEffect]>, BASE_HLO_SelectOp { } def HLO_ReverseOp: HLO_Op<"reverse", - [NoSideEffect, SameOperandsAndResultElementType]>, BASE_HLO_ReverseOp { + [NoSideEffect, SameOperandsAndResultType]>, BASE_HLO_ReverseOp { let arguments = (ins HLO_Tensor:$operand, I64ElementsAttr:$dimensions