Skip to content

Commit 03aae4f

Browse files
authored
[MLIR][TOSA] Update CustomOp input and output names (#118408)
Update input name from input to input_list Update output name from outputs to output_list To match specification: https://www.mlplatform.org/tosa/tosa_spec.html#_custom Signed-off-by: Jerry Ge <[email protected]>
1 parent 5cbc421 commit 03aae4f

File tree

1 file changed

+4
-4
lines changed

1 file changed

+4
-4
lines changed

mlir/include/mlir/Dialect/Tosa/IR/TosaOps.td

Lines changed: 4 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -2013,9 +2013,9 @@ def Tosa_CustomOp : Tosa_Op<"custom"> {
20132013
`implementation_attrs` is a string which is a backend and identifier specific
20142014
set of attributes to the custom operator.
20152015

2016-
`inputs` is the set of tensor inputs to the custom operator.
2016+
`input_list` is the set of tensor inputs to the custom operator.
20172017

2018-
`outputs is the list of tensors returned by the operator. The number of operators
2018+
`output_list` is the list of tensors returned by the operator. The number of operators
20192019
is backend specific.
20202020

20212021
Example:
@@ -2031,11 +2031,11 @@ def Tosa_CustomOp : Tosa_Op<"custom"> {
20312031
StrAttr:$operator_name,
20322032
StrAttr:$domain_name,
20332033
StrAttr:$implementation_attrs,
2034-
Variadic<Tosa_Tensor>:$inputs
2034+
Variadic<Tosa_Tensor>:$input_list
20352035
);
20362036

20372037
let results = (outs
2038-
Variadic<Tosa_Tensor>:$outputs
2038+
Variadic<Tosa_Tensor>:$output_list
20392039
);
20402040

20412041
let assemblyFormat = "operands attr-dict `:` functional-type(operands, results)";

0 commit comments

Comments
 (0)