@@ -228,20 +228,20 @@ struct LinearizeVectorExtractStridedSlice final
228228 " strides other than 1 not supported" );
229229 }
230230
231- ArrayRef<int64_t > inputShape =
232- extractStridedSliceOp.getSourceVectorType ().getShape ();
233-
234- ArrayRef<int64_t > outputType = extractStridedSliceOp.getType ().getShape ();
235-
236- auto maybeIntOffsets =
231+ FailureOr<SmallVector<int64_t >> offsets =
237232 intsFromArrayAttr (extractStridedSliceOp.getOffsets ());
238- if (failed (maybeIntOffsets )) {
233+ if (failed (offsets )) {
239234 return rewriter.notifyMatchFailure (extractStridedSliceOp,
240235 " failed to get integer offsets" );
241236 }
242237
238+ ArrayRef<int64_t > inputShape =
239+ extractStridedSliceOp.getSourceVectorType ().getShape ();
240+
241+ ArrayRef<int64_t > outputShape = extractStridedSliceOp.getType ().getShape ();
242+
243243 SmallVector<int64_t > indices = getFlattenedStridedSliceIndices (
244- outputType , inputShape, maybeIntOffsets .value ());
244+ outputShape , inputShape, offsets .value ());
245245
246246 Value srcVector = adaptor.getVector ();
247247 rewriter.replaceOpWithNewOp <vector::ShuffleOp>(
@@ -299,13 +299,14 @@ struct LinearizeVectorInsertStridedSlice final
299299 ArrayRef<int64_t > outputShape = outputType.getShape ();
300300 int64_t nOutputElements = outputType.getNumElements ();
301301
302- auto maybeIntOffsets = intsFromArrayAttr (insertStridedSliceOp.getOffsets ());
303- if (failed (maybeIntOffsets)) {
302+ FailureOr<SmallVector<int64_t >> offsets =
303+ intsFromArrayAttr (insertStridedSliceOp.getOffsets ());
304+ if (failed (offsets)) {
304305 return rewriter.notifyMatchFailure (insertStridedSliceOp,
305306 " failed to get integer offsets" );
306307 }
307308 SmallVector<int64_t > sliceIndices = getFlattenedStridedSliceIndices (
308- inputShape, outputShape, maybeIntOffsets .value ());
309+ inputShape, outputShape, offsets .value ());
309310
310311 SmallVector<int64_t > indices (nOutputElements, 0 );
311312 std::iota (indices.begin (), indices.end (), 0 );
0 commit comments