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

[MLIR][HLO] Use `#` operator for list concatenation in C/HLO definitions 

PiperOrigin-RevId: 373110780
This commit is contained in:
A. Unique TensorFlower 2021-05-11 02:52:35 -07:00 committed by TensorFlow MLIR Team
parent 920f261262
commit 6191b3e528
1 changed files with 5 additions and 7 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

12
include/mlir-hlo/Dialect/mhlo/IR/chlo_ops.td
View File

@ -76,11 +76,9 @@ class HLOClient_Op<string mnemonic, list<OpTrait> traits> :
//===----------------------------------------------------------------------===// //===----------------------------------------------------------------------===//
class HLOClient_BroadcastBinaryElementwiseOp< class HLOClient_BroadcastBinaryElementwiseOp<
string mnemonic, list<OpTrait> traits> : string mnemonic, list<OpTrait> traits> : HLOClient_Op<mnemonic, traits # [
HLOClient_Op<mnemonic, DeclareOpInterfaceMethods<InferShapedTypeOpInterface, [
!listconcat(traits, [ "inferReturnTypeComponents", "reifyReturnTypeShapes"]>]> {
DeclareOpInterfaceMethods<InferShapedTypeOpInterface,
["inferReturnTypeComponents", "reifyReturnTypeShapes"]>])> {
let arguments = (ins let arguments = (ins
HLO_Tensor:$lhs, HLO_Tensor:$lhs,
HLO_Tensor:$rhs, HLO_Tensor:$rhs,
@ -411,8 +409,8 @@ def HLOClient_BroadcastComplexOp : HLOClient_BroadcastBinaryElementwiseOp<
class HLOClient_UnaryElementwiseOp<string mnemonic, list<OpTrait> traits, class HLOClient_UnaryElementwiseOp<string mnemonic, list<OpTrait> traits,
Type ArgTensorType, Type ResultTensorType> : HLOClient_Op<mnemonic, Type ArgTensorType, Type ResultTensorType> : HLOClient_Op<mnemonic,
!listconcat(traits, [InferFusibilityOpInterface, NoSideEffect, traits # [InferFusibilityOpInterface, NoSideEffect,
SameOperandsAndResultShape])> { SameOperandsAndResultShape]> {
let arguments = (ins ArgTensorType:$operand); let arguments = (ins ArgTensorType:$operand);
let results = (outs ResultTensorType:$result); let results = (outs ResultTensorType:$result);