Searched refs:vectorOffsets (Results 1 – 4 of 4) sorted by relevance
24 SmallVector<int64_t, 4> vectorOffsets(rank); in delinearize() local27 vectorOffsets[r] = index / sliceStrides[r]; in delinearize()30 return vectorOffsets; in delinearize()
70 ArrayRef<int64_t> sizes, ArrayRef<int64_t> vectorOffsets) { in computeElementOffsetsFromVectorSliceOffsets() argument72 for (auto it : llvm::zip(vectorOffsets, sizes)) in computeElementOffsetsFromVectorSliceOffsets()
35 SmallVector<int64_t, 4> vectorOffsets = delinearize(dstSliceStrides, index); in getVectorOffset() local37 computeElementOffsetsFromVectorSliceOffsets(targetShape, vectorOffsets); in getVectorOffset()83 SmallVector<int64_t> vectorOffsets(sliceStrides.size()); in delinearize() local85 vectorOffsets[idx] = index / sliceStrides[idx]; in delinearize()88 return vectorOffsets; in delinearize()96 SmallVector<int64_t> vectorOffsets = delinearize(index); in getVectorOffset() local98 computeElementOffsetsFromVectorSliceOffsets(vectorShape, vectorOffsets); in getVectorOffset()
52 ArrayRef<int64_t> vectorOffsets);