refactor(ONNX): extracts inputTensor dimensions from rank derivation

llvm · Jan 8, 2025 · 7d8d04b · 7d8d04b
1 parent 7f3913d
commit 7d8d04b
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/lib/Conversion/TorchOnnxToTorch/DefaultDomainQtoZ.cpp b/lib/Conversion/TorchOnnxToTorch/DefaultDomainQtoZ.cpp
@@ -2752,7 +2752,8 @@ void mlir::torch::onnx_c::populateDefaultDomainQtoZ(
         Value inputTensor = operands[0];
         Torch::ValueTensorType inputTensor_shape =
             cast<Torch::ValueTensorType>(inputTensor.getType());
-        unsigned rank = inputTensor_shape.getSizes().size();
+        ArrayRef<int64_t> inputTensor_dimensions = inputTensor_shape.getSizes();
+        unsigned rank = inputTensor_dimensions.size();
 
         Value cstFalse =
             rewriter.create<Torch::ConstantBoolOp>(binder.getLoc(), false);