Searched refs:sliceStrides (Results 1 – 3 of 3) sorted by relevance
21 llvm::SmallVector<int64_t, 4> mlir::delinearize(ArrayRef<int64_t> sliceStrides, in delinearize() argument23 int64_t rank = sliceStrides.size(); in delinearize()26 assert(sliceStrides[r] > 0); in delinearize()27 vectorOffsets[r] = index / sliceStrides[r]; in delinearize()28 index %= sliceStrides[r]; in delinearize()
62 SmallVector<int64_t, 4> sliceStrides(rank); in computeStrides() local63 sliceStrides[rank - 1] = 1; in computeStrides()65 sliceStrides[r] = sliceStrides[r + 1] * sliceDimCounts[r + 1]; in computeStrides()66 return sliceStrides; in computeStrides()
49 SmallVector<int64_t> sliceStrides; member in __anon77a0e1ec0111::DecomposeShapeIterator58 sliceStrides(originalShape.size()) { in DecomposeShapeIterator()73 sliceStrides[idx] = accum; in DecomposeShapeIterator()83 SmallVector<int64_t> vectorOffsets(sliceStrides.size()); in delinearize()85 vectorOffsets[idx] = index / sliceStrides[idx]; in delinearize()86 index %= sliceStrides[idx]; in delinearize()