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: // 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 // = pi if x == -1
def : Pat<(HLOClient_AcosOp $input), def : Pat<(HLOClient_AcosOp $input),
(HLO_SelectOp (HLO_SelectOp
(HLO_CompareOp $input, (HLO_CompareOp
(HLO_ConstantLike<"0"> $input), $input,
(HLO_ConstantLike<"-1"> $input),
HLO_COMPARISON_DIRECTION_NE HLO_COMPARISON_DIRECTION_NE
), ),
(HLO_MulOp (HLO_MulOp
(HLO_ConstantLike<"2.0f"> $input), (HLO_ConstantLike<"2"> $input),
(HLO_Atan2Op (HLO_Atan2Op
(HLO_SqrtOp (HLO_SqrtOp
(HLO_SubOp (HLO_SubOp
@ -47,7 +48,8 @@ def : Pat<(HLOClient_AcosOp $input),
) )
) )
), ),
(HLO_ConstantLike<"M_PI"> $input))>; (HLO_ConstantLike<"M_PI"> $input)
)>;
// Express `atan` as // Express `atan` as
// atan(x) = atan2(x, 1) // atan(x) = atan2(x, 1)