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

Make last argument of conv variadic.

This commit is contained in:
Doru Bercea 2020-01-14 11:17:52 -05:00
parent deef363309
commit fc352745e0
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
src/dialect/onnx/onnxop.inc
View File

@ -322,7 +322,7 @@ def ONNXConvOp:ONNX_Op<"Conv",
"The convolution operator consumes an input tensor and a filter, and"
"computes the output."
}];
let arguments = (ins AnyTypeOf<[AnyMemRef, AnyTensor]>:$X, AnyTypeOf<[AnyMemRef, AnyTensor]>:$W, AnyTypeOf<[AnyMemRef, AnyTensor]>:$B);
let arguments = (ins AnyTypeOf<[AnyMemRef, AnyTensor]>:$X, AnyTypeOf<[AnyMemRef, AnyTensor]>:$W, Variadic<AnyTypeOf<[AnyMemRef, AnyTensor]>>:$B);
let results = (outs AnyTypeOf<[AnyMemRef, AnyTensor]>);
}