Skip to content

Commit 7ac2575

Browse files
committed
refactor(ONNX): renames resultType to outputTensor_type
- emphasizes parallel to `inputTensor_type`
1 parent 69b238f commit 7ac2575

File tree

1 file changed

+3
-3
lines changed

1 file changed

+3
-3
lines changed

Diff for: lib/Conversion/TorchOnnxToTorch/DefaultDomainQtoZ.cpp

+3-3
Original file line numberDiff line numberDiff line change
@@ -2686,7 +2686,7 @@ void mlir::torch::onnx_c::populateDefaultDomainQtoZ(
26862686
});
26872687
patterns.onOp(
26882688
"Resize", 11, [](OpBinder binder, ConversionPatternRewriter &rewriter) {
2689-
Torch::ValueTensorType resultType;
2689+
Torch::ValueTensorType outputTensor_type;
26902690
llvm::SmallVector<Value> operands;
26912691
std::string mode, nearest_mode, coordTfMode;
26922692
int64_t antialias, exclude_outside;
@@ -2705,7 +2705,7 @@ void mlir::torch::onnx_c::populateDefaultDomainQtoZ(
27052705
}
27062706

27072707
if (binder.tensorOperandsList(operands) ||
2708-
binder.tensorResultType(resultType) ||
2708+
binder.tensorResultType(outputTensor_type) ||
27092709
binder.customOpNameStringAttr(mode, "mode", "nearest") ||
27102710
binder.customOpNameStringAttr(
27112711
coordTfMode, "coordinate_transformation_mode", "half_pixel") ||
@@ -2829,7 +2829,7 @@ void mlir::torch::onnx_c::populateDefaultDomainQtoZ(
28292829
}
28302830
rewriter
28312831
.replaceOpWithNewOp<Torch::Aten__InterpolateSizeListScaleListOp>(
2832-
binder.op, resultType, inputTensor, sizesValueList,
2832+
binder.op, outputTensor_type, inputTensor, sizesValueList,
28332833
scalesValueList, modeStrValue,
28342834
/* AnyTorchOptionalBoolType:$align_corners */ alignCorners,
28352835
/* AnyTorchOptionalBoolType:$recompute_scale_factor */ noneVal,

0 commit comments

Comments
 (0)