@@ -2690,7 +2690,7 @@ void mlir::torch::onnx_c::populateDefaultDomainQtoZ(
2690
2690
});
2691
2691
patterns.onOp (
2692
2692
" Resize" , 11 , [](OpBinder binder, ConversionPatternRewriter &rewriter) {
2693
- Torch::ValueTensorType resultType ;
2693
+ Torch::ValueTensorType typeOfOutputTensor ;
2694
2694
llvm::SmallVector<Value> operands;
2695
2695
std::string mode, nearest_mode, coordTfMode;
2696
2696
int64_t antialias, exclude_outside;
@@ -2709,7 +2709,7 @@ void mlir::torch::onnx_c::populateDefaultDomainQtoZ(
2709
2709
}
2710
2710
2711
2711
if (binder.tensorOperandsList (operands) ||
2712
- binder.tensorResultType (resultType ) ||
2712
+ binder.tensorResultType (typeOfOutputTensor ) ||
2713
2713
binder.customOpNameStringAttr (mode, " mode" , " nearest" ) ||
2714
2714
binder.customOpNameStringAttr (
2715
2715
coordTfMode, " coordinate_transformation_mode" , " half_pixel" ) ||
@@ -2833,7 +2833,7 @@ void mlir::torch::onnx_c::populateDefaultDomainQtoZ(
2833
2833
}
2834
2834
rewriter
2835
2835
.replaceOpWithNewOp <Torch::Aten__InterpolateSizeListScaleListOp>(
2836
- binder.op , resultType , inputTensor, sizesValueList,
2836
+ binder.op , typeOfOutputTensor , inputTensor, sizesValueList,
2837
2837
scalesValueList, modeStrValue,
2838
2838
/* AnyTorchOptionalBoolType:$align_corners */ alignCorners,
2839
2839
/* AnyTorchOptionalBoolType:$recompute_scale_factor */ noneVal,
0 commit comments