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

Add SameOperandsAndResultType trait to SignOp and ReverseOp in HLO

Browse Source 
Currently, these ops are using SameOperandsAndResultShape and SameOperandsAndResultElementType, respectively.

PiperOrigin-RevId: 270364879
This commit is contained in:
Smit Hinsu 2019-09-20 15:37:42 -07:00 committed by TensorFlower Gardener
parent 9f2eec8221
commit 9877c212c8
1 changed files with 2 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
tensorflow/compiler/mlir/xla/ir/hlo_ops.td
View File

@ -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_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", def HLO_TanhOp: HLO_UnaryElementwiseOp<"tanh",
[ResultsAreFloatLike, NoSideEffect, SameOperandsAndResultType]>, BASE_HLO_TanhOp; [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", def HLO_ReverseOp: HLO_Op<"reverse",
[NoSideEffect, SameOperandsAndResultElementType]>, BASE_HLO_ReverseOp { [NoSideEffect, SameOperandsAndResultType]>, BASE_HLO_ReverseOp {
let arguments = (ins let arguments = (ins
HLO_Tensor:$operand, HLO_Tensor:$operand,
I64ElementsAttr:$dimensions I64ElementsAttr:$dimensions