diff --git a/lib/Conversion/TorchOnnxToTorch/DefaultDomainGtoP.cpp b/lib/Conversion/TorchOnnxToTorch/DefaultDomainGtoP.cpp index ad6c91e40..e8ac725fc 100644 --- a/lib/Conversion/TorchOnnxToTorch/DefaultDomainGtoP.cpp +++ b/lib/Conversion/TorchOnnxToTorch/DefaultDomainGtoP.cpp @@ -124,11 +124,11 @@ void mlir::torch::onnx_c::populateDefaultDomainGtoP( return rewriter.notifyMatchFailure(binder.op, "gridShape[3] expected to be 2"); std::string mode; - if (binder.customOpNameStringAttr(mode, "mode", "bilinear")) + if (binder.customOpNameStringAttr(mode, "mode", "linear")) return rewriter.notifyMatchFailure(binder.op, "mode bind failure"); - if (mode != "bilinear") + if (mode != "linear" && mode != "bilinear") return rewriter.notifyMatchFailure( - binder.op, "currently only mode : bilinear supported"); + binder.op, "currently only mode : linear supported"); std::string padding; if (binder.customOpNameStringAttr(padding, "padding_mode", "zeros")) return rewriter.notifyMatchFailure(binder.op,