llvm
/
torch-mlir
mirror of https://github.com/llvm/torch-mlir
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

reduce some min opset versions

pull/3877/head
zjgarvey 2024-11-14 22:28:36 -06:00
parent 06d17897f0
commit 4ebe197de9
1 changed files with 2 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
lib/Conversion/TorchOnnxToTorch/DefaultDomainGtoP.cpp
View File

@ -1524,7 +1524,7 @@ void mlir::torch::onnx_c::populateDefaultDomainGtoP(
return success(); return success();
}); });
patterns.onOp("Greater", 16, patterns.onOp("Greater", 7,
[](OpBinder binder, ConversionPatternRewriter &rewriter) { [](OpBinder binder, ConversionPatternRewriter &rewriter) {
Torch::ValueTensorType resultType; Torch::ValueTensorType resultType;
Value lhs, rhs; Value lhs, rhs;
@ -1931,7 +1931,7 @@ void mlir::torch::onnx_c::populateDefaultDomainGtoP(
return success(); return success();
}); });
patterns.onOp( patterns.onOp(
"Gather", 13, [](OpBinder binder, ConversionPatternRewriter &rewriter) { "Gather", 1, [](OpBinder binder, ConversionPatternRewriter &rewriter) {
Torch::ValueTensorType resultType; Torch::ValueTensorType resultType;
Value data, indices; Value data, indices;
int64_t axis; int64_t axis;