Searched refs:vectorOffsets (Results 1 – 4 of 4) sorted by relevance
| /external/llvm-project/mlir/lib/Dialect/Vector/ |
| D | VectorUtils.cpp | 84 SmallVector<int64_t, 4> vectorOffsets(rank); in delinearize() local
87 vectorOffsets[r] = index / sliceStrides[r]; in delinearize()
90 return vectorOffsets; in delinearize()
94 ArrayRef<int64_t> sizes, ArrayRef<int64_t> vectorOffsets) { in computeElementOffsetsFromVectorSliceOffsets() argument
96 for (auto it : llvm::zip(vectorOffsets, sizes)) in computeElementOffsetsFromVectorSliceOffsets()
|
| D | VectorTransforms.cpp | 211 auto vectorOffsets = delinearize(sliceStrides, i); in generateExtractSlicesOpResultType() local
213 computeElementOffsetsFromVectorSliceOffsets(sizes, vectorOffsets); in generateExtractSlicesOpResultType()
265 ArrayRef<int64_t> vectorOffsets, in getUnrolledVectorLinearIndex() argument
269 getMappedElements(indexMap, vectorOffsets, sliceOffsets); in getUnrolledVectorLinearIndex()
278 Location loc, UnrolledVectorState &state, ArrayRef<int64_t> vectorOffsets, in getOrCreateUnrolledVectorSlice() argument
288 getUnrolledVectorLinearIndex(state, vectorOffsets, indexMap); in getOrCreateUnrolledVectorSlice()
407 auto vectorOffsets = delinearize(sliceStrides, i); in unrollSingleResultStructuredOp() local
409 computeElementOffsetsFromVectorSliceOffsets(targetShape, vectorOffsets); in unrollSingleResultStructuredOp()
419 op->getLoc(), unrolledVectorState[i], vectorOffsets, elementOffsets, in unrollSingleResultStructuredOp()
430 resultValueState, vectorOffsets, vectors[resultIndex].indexMap); in unrollSingleResultStructuredOp()
[all …]
|
| D | VectorOps.cpp | 958 auto vectorOffsets = delinearize(sliceStrides, i); in isValidExtractOrInsertSlicesType() local
960 computeElementOffsetsFromVectorSliceOffsets(sizes, vectorOffsets); in isValidExtractOrInsertSlicesType()
|
| /external/llvm-project/mlir/include/mlir/Dialect/Vector/ |
| D | VectorUtils.h | 58 ArrayRef<int64_t> vectorOffsets);
|