colin
/
mlir-hlo
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

[MLIR][KernelGen] Add E2E test for `tf.Acos` 

PiperOrigin-RevId: 335419310
This commit is contained in:
A. Unique TensorFlower 2020-10-05 08:13:58 -07:00 committed by TensorFlow MLIR Team
parent bae0815ef0
commit a33441907c
1 changed files with 7 additions and 5 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

12
lib/Dialect/mhlo/transforms/chlo_legalize_to_hlo_patterns.td
View File

@ -24,16 +24,17 @@ include "mlir-hlo/Dialect/mhlo/IR/chlo_ops.td"
//===----------------------------------------------------------------------===//
// Expand acos to MHLO dialect as follows:
// acos(x) = 2 * atan(sqrt(1 - x^2) / (1 + x)) if x != -1
// acos(x) = 2 * atan2(sqrt(1 - x^2), (1 + x)) if x != -1
// = pi if x == -1
def : Pat<(HLOClient_AcosOp $input),
(HLO_SelectOp
(HLO_CompareOp $input,
(HLO_ConstantLike<"0"> $input),
(HLO_CompareOp
$input,
(HLO_ConstantLike<"-1"> $input),
HLO_COMPARISON_DIRECTION_NE
),
(HLO_MulOp
(HLO_ConstantLike<"2.0f"> $input),
(HLO_ConstantLike<"2"> $input),
(HLO_Atan2Op
(HLO_SqrtOp
(HLO_SubOp
@ -47,7 +48,8 @@ def : Pat<(HLOClient_AcosOp $input),
)
)
),
(HLO_ConstantLike<"M_PI"> $input))>;
(HLO_ConstantLike<"M_PI"> $input)
)>;
// Express `atan` as
// atan(x) = atan2(x, 1)