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

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]>);
}