Lines Matching refs:elementOffsets
36 SmallVector<int64_t, 4> elementOffsets = in getVectorOffset() local
38 return elementOffsets; in getVectorOffset()
97 SmallVector<int64_t> elementOffsets = in getVectorOffset() local
99 return elementOffsets; in getVectorOffset()
105 static SmallVector<Value> sliceTransferIndices(ArrayRef<int64_t> elementOffsets, in sliceTransferIndices() argument
123 getAffineConstantExpr(elementOffsets[dim.index()], ctx); in sliceTransferIndices()
215 SmallVector<int64_t, 4> elementOffsets = in matchAndRewrite() local
218 sliceTransferIndices(elementOffsets, originalIndices, in matchAndRewrite()
226 loc, slicedRead, result, elementOffsets, strides); in matchAndRewrite()
266 SmallVector<int64_t, 4> elementOffsets = in matchAndRewrite() local
269 loc, writeOp.getVector(), elementOffsets, *targetShape, strides); in matchAndRewrite()
271 sliceTransferIndices(elementOffsets, originalIndices, in matchAndRewrite()
797 SmallVector<int64_t, 4> elementOffsets = in matchAndRewrite() local
799 SmallVector<int64_t, 4> permutedOffsets(elementOffsets.size()); in matchAndRewrite()
800 SmallVector<int64_t, 4> permutedShape(elementOffsets.size()); in matchAndRewrite()
803 permutedOffsets[indices.value()] = elementOffsets[indices.index()]; in matchAndRewrite()
811 loc, tranposedSlice, result, elementOffsets, strides); in matchAndRewrite()