Skip to content

Commit 9b766e2

Browse files
committed
refactor(ONNX): renames resultType to typeOfOutputTensor
- emphasizes parallel to `typeOfInputTensor`
1 parent 379ddc9 commit 9b766e2

File tree

1 file changed

+3
-3
lines changed

1 file changed

+3
-3
lines changed

lib/Conversion/TorchOnnxToTorch/DefaultDomainQtoZ.cpp

+3-3
Original file line numberDiff line numberDiff line change
@@ -2690,7 +2690,7 @@ void mlir::torch::onnx_c::populateDefaultDomainQtoZ(
26902690
});
26912691
patterns.onOp(
26922692
"Resize", 11, [](OpBinder binder, ConversionPatternRewriter &rewriter) {
2693-
Torch::ValueTensorType resultType;
2693+
Torch::ValueTensorType typeOfOutputTensor;
26942694
llvm::SmallVector<Value> operands;
26952695
std::string mode, nearest_mode, coordTfMode;
26962696
int64_t antialias, exclude_outside;
@@ -2709,7 +2709,7 @@ void mlir::torch::onnx_c::populateDefaultDomainQtoZ(
27092709
}
27102710

27112711
if (binder.tensorOperandsList(operands) ||
2712-
binder.tensorResultType(resultType) ||
2712+
binder.tensorResultType(typeOfOutputTensor) ||
27132713
binder.customOpNameStringAttr(mode, "mode", "nearest") ||
27142714
binder.customOpNameStringAttr(
27152715
coordTfMode, "coordinate_transformation_mode", "half_pixel") ||
@@ -2833,7 +2833,7 @@ void mlir::torch::onnx_c::populateDefaultDomainQtoZ(
28332833
}
28342834
rewriter
28352835
.replaceOpWithNewOp<Torch::Aten__InterpolateSizeListScaleListOp>(
2836-
binder.op, resultType, inputTensor, sizesValueList,
2836+
binder.op, typeOfOutputTensor, inputTensor, sizesValueList,
28372837
scalesValueList, modeStrValue,
28382838
/* AnyTorchOptionalBoolType:$align_corners */ alignCorners,
28392839
/* AnyTorchOptionalBoolType:$recompute_scale_factor */ noneVal,

0 commit comments

Comments
 (0)