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

Add SameOperandsAndResultShape trait to ConvertOp and IsFiniteOp. 

PiperOrigin-RevId: 354070005
This commit is contained in:
Adrian Kuegel 2021-01-27 05:09:01 -08:00 committed by TensorFlow MLIR Team
parent fa059259bc
commit f4f728f18e
1 changed files with 3 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

5
include/mlir-hlo/Dialect/mhlo/IR/hlo_ops.td
View File

@ -160,7 +160,8 @@ def HLO_CeilOp: HLO_UnaryElementwiseOp<"ceil",
[NoSideEffect, SameOperandsAndResultType], HLO_FpTensor>, BASE_HLO_CeilOp;
def HLO_ConvertOp : HLO_UnaryElementwiseOp<"convert",
[NoSideEffect], HLO_Tensor>, BASE_HLO_ConvertOp {
[NoSideEffect, SameOperandsAndResultShape], HLO_Tensor>,
BASE_HLO_ConvertOp {
let builders = [
OpBuilderDAG<(ins "Value":$operand, "Type":$result_element_ty)>];
@ -196,7 +197,7 @@ def HLO_ImagOp: HLO_UnaryElementwiseOp<"imag",
}
def HLO_IsFiniteOp: HLO_UnaryElementwiseOp<"is_finite",
[NoSideEffect], HLO_Tensor>,
[NoSideEffect, SameOperandsAndResultShape], HLO_Tensor>,
BASE_HLO_IsFiniteOp {
let arguments = (ins HLO_FpTensor:$x);
let results = (outs HLO_PredTensor:$y);