Searched refs:sliceDimCounts (Results 1 – 2 of 2) sorted by relevance
62 SmallVector<int64_t, 4> sliceDimCounts(rank); in computeStrides() local64 sliceDimCounts[r] = ceilDiv(shape[r], sizes[r]); in computeStrides()69 sliceStrides[r] = sliceStrides[r + 1] * sliceDimCounts[r + 1]; in computeStrides()
204 auto sliceDimCounts = *maybeDimSliceCounts; in generateExtractSlicesOpResultType() local207 auto sliceStrides = computeStrides(sliceDimCounts); in generateExtractSlicesOpResultType()208 int64_t sliceCount = computeMaxLinearIndex(sliceDimCounts); in generateExtractSlicesOpResultType()522 auto sliceDimCounts = *maybeDimSliceCounts; in generateTransferOpSlices() local523 auto sliceStrides = computeStrides(sliceDimCounts); in generateTransferOpSlices()