Skip to content

Commit d8b5290

Browse files
authored
[ONNX] reduce min opset versions for Greater and Gather (#3877)
No appreciable difference between the lower opset version and the newer one. See <https://onnx.ai/onnx/operators/onnx__Greater.html> and <https://onnx.ai/onnx/operators/onnx__Gather.html>
1 parent 66dc639 commit d8b5290

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

lib/Conversion/TorchOnnxToTorch/DefaultDomainGtoP.cpp

+2-2
Original file line numberDiff line numberDiff line change
@@ -1541,7 +1541,7 @@ void mlir::torch::onnx_c::populateDefaultDomainGtoP(
15411541

15421542
return success();
15431543
});
1544-
patterns.onOp("Greater", 16,
1544+
patterns.onOp("Greater", 7,
15451545
[](OpBinder binder, ConversionPatternRewriter &rewriter) {
15461546
Torch::ValueTensorType resultType;
15471547
Value lhs, rhs;
@@ -1948,7 +1948,7 @@ void mlir::torch::onnx_c::populateDefaultDomainGtoP(
19481948
return success();
19491949
});
19501950
patterns.onOp(
1951-
"Gather", 13, [](OpBinder binder, ConversionPatternRewriter &rewriter) {
1951+
"Gather", 1, [](OpBinder binder, ConversionPatternRewriter &rewriter) {
19521952
Torch::ValueTensorType resultType;
19531953
Value data, indices;
19541954
int64_t axis;

0 commit comments

Comments
 (0)