We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
1 parent f42c7e4 commit 73a4afbCopy full SHA for 73a4afb
1 file changed
lib/Conversion/TorchToTosa/TorchToTosa.cpp
@@ -4019,7 +4019,7 @@ LogicalResult ConvertAtenOp<AtenIndexSelectOp>::matchAndRewrite(
4019
4020
auto index = adaptor.getIndex();
4021
auto indexType = dyn_cast<RankedTensorType>(index.getType());
4022
- auto indexShape = indexType.getShape();
+ auto indexShape = SmallVector<int64_t>(indexType.getShape());
4023
4024
if (!indexType)
4025
return rewriter.notifyMatchFailure(
0 commit comments