Searched refs:sliceStrides (Results 1 – 3 of 3) sorted by relevance
| /external/llvm-project/mlir/lib/Dialect/Vector/ |
| D | VectorUtils.cpp | 66 SmallVector<int64_t, 4> sliceStrides(rank); in computeStrides() local
67 sliceStrides[rank - 1] = 1; in computeStrides()
69 sliceStrides[r] = sliceStrides[r + 1] * sliceDimCounts[r + 1]; in computeStrides()
70 return sliceStrides; in computeStrides()
81 SmallVector<int64_t, 4> mlir::delinearize(ArrayRef<int64_t> sliceStrides, in delinearize() argument
83 int64_t rank = sliceStrides.size(); in delinearize()
86 assert(sliceStrides[r] > 0); in delinearize()
87 vectorOffsets[r] = index / sliceStrides[r]; in delinearize()
88 index %= sliceStrides[r]; in delinearize()
|
| D | VectorTransforms.cpp | 207 auto sliceStrides = computeStrides(sliceDimCounts); in generateExtractSlicesOpResultType() local
211 auto vectorOffsets = delinearize(sliceStrides, i); in generateExtractSlicesOpResultType()
285 SmallVector<int64_t, 4> sliceStrides(state.unrolledShape.size(), 1); in getOrCreateUnrolledVectorSlice() local
394 auto sliceStrides = computeStrides(unrollFactors); in unrollSingleResultStructuredOp() local
407 auto vectorOffsets = delinearize(sliceStrides, i); in unrollSingleResultStructuredOp()
523 auto sliceStrides = computeStrides(sliceDimCounts); in generateTransferOpSlices() local
548 auto vectorOffsets = delinearize(sliceStrides, i); in generateTransferOpSlices()
846 auto sliceStrides = computeStrides( in getProducerValue() local
851 auto vectorOffsets = delinearize(sliceStrides, tupleIndex); in getProducerValue()
870 auto sliceStrides = computeStrides( in getProducerValue() local
[all …]
|
| D | VectorOps.cpp | 956 auto sliceStrides = computeStrides(shape, sizes); in isValidExtractOrInsertSlicesType() local
958 auto vectorOffsets = delinearize(sliceStrides, i); in isValidExtractOrInsertSlicesType()
|