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

Allow tuple as results of mhlo.custom_call op (NFC) 

PiperOrigin-RevId: 376009518
This commit is contained in:
Mehdi Amini 2021-05-26 12:55:03 -07:00 committed by TensorFlow MLIR Team
parent 26a0053d7d
commit af01c08ce6
1 changed files with 2 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

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

@ -1471,12 +1471,12 @@ def HLO_CustomCallOp: HLO_Op<"custom_call", []> {
See https://www.tensorflow.org/xla/operation_semantics#customcall. See https://www.tensorflow.org/xla/operation_semantics#customcall.
}]; }];
let arguments = (ins let arguments = (ins
Variadic<HLO_Tensor>:$args, Variadic<HLO_TensorOrTuple>:$args,
StrAttr:$call_target_name, StrAttr:$call_target_name,
DefaultValuedAttr<BoolAttr, "false">:$has_side_effect, DefaultValuedAttr<BoolAttr, "false">:$has_side_effect,
DefaultValuedAttr<StrAttr, "">:$backend_config DefaultValuedAttr<StrAttr, "">:$backend_config
); );
let results = (outs Variadic<HLO_Tensor>); let results = (outs Variadic<HLO_TensorOrTuple>);
let hasCustomHLOConverter = 1; let hasCustomHLOConverter = 1;
} }