We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
1 parent 9188846 commit 0f5dbbeCopy full SHA for 0f5dbbe
lib/Conversion/TorchOnnxToTorch/DefaultDomainGtoP.cpp
@@ -180,7 +180,7 @@ void mlir::torch::onnx_c::populateDefaultDomainGtoP(
180
181
auto conditionType =
182
cast<Torch::ValueTensorType>(conditionTensor.getType());
183
- if (!conditionType || conditionType.getSizes().size() > 1)
+ if (!conditionType || conditionType.getSizes().size() != 1)
184
return rewriter.notifyMatchFailure(
185
binder.op, "condition must have one single element per "
186
"https://onnx.ai/onnx/operators/onnx__If.html");
0 commit comments