Lines Matching refs:i

52     for (int i = origRank - 1; i >= 0; i--) {  in getFlattenedIndex()  local
53 index += uint64_t(indices[i] * subElems); in getFlattenedIndex()
54 subElems *= shape[i]; in getFlattenedIndex()
97 size_t i = 0; in populate() local
99 for (; i < origIdx.size(); i++) { in populate()
100 int origDim = traversalOrder[i]; in populate()
101 origIdx[origDim] = (*indices)[i]; in populate()
104 for (; i < (*indices).size(); i++) { in populate()
105 const int blockIdx = traversalOrder[i] - origRank; in populate()
107 origIdx[origDim] = origIdx[origDim] * blockSize[blockIdx] + (*indices)[i]; in populate()
116 for (int i = 0; i < shapeOfLevel; i++) { in populate() local
117 (*indices)[level] = i; in populate()
118 populate(srcData, indices, level + 1, prevIdx * shapeOfLevel + i, destData, destDims, in populate()
124 for (int i = arraySegments[prevIdx]; i < arraySegments[prevIdx + 1]; i++) { in populate() local
125 (*indices)[level] = arrayIndices[i]; in populate()
126 populate(srcData, indices, level + 1, i, destData, destDims, dimFormat, traversalOrder, in populate()
159 for (uint32_t i = kInputArrSeg; i < inputCount; i++) { in densify() local
160 inputShapes.push_back(context->getInputShape(i)); in densify()
161 const int32_t* metadata = context->getInputBuffer<int32_t>(i); in densify()
171 for (size_t i = 0; i < dimFormat.size(); i++) { in densify() local
172 dimFormat[i] = dimFormatPtr[i]; in densify()
173 dimensions[i] = dimensionsPtr[i]; in densify()
174 if (dimFormat[i] == 0) { in densify()
175 dimMetadata[i * 2] = {dimensions[i]}; in densify()
177 dimMetadata[i * 2] = // array segments in densify()
178 arrToVector(dimMetadataPtrs[i * 2], in densify()
179 inputShapes[i * 2 + kInputArrSeg].dimensions.front()); in densify()
180 dimMetadata[i * 2 + 1] = // array indices in densify()
181 arrToVector(dimMetadataPtrs[i * 2 + 1], in densify()
182 inputShapes[i * 2 + kInputArrIdx].dimensions.front()); in densify()
190 for (uint32_t i = 0; i < inputShapes[kInputBlockMap].dimensions.front(); i++) { in densify() local
191 const int32_t origDim = traversalOrder[origRank + i]; in densify()
192 blockSize[i] = dimensions[origDim]; in densify()
208 for (size_t i = 0; i < denseTotal; i++) { in densify() local
209 destData[i] = zeroPoint; in densify()
231 size_t i = 0; in prepare() local
232 for (; i < destDims.size(); i++) { in prepare()
233 const int32_t origDim = traversalOrder[i]; in prepare()
234 destDims[origDim] = dimensions[i]; in prepare()
236 for (; i < dimensionsShape.dimensions.front(); i++) { in prepare()
237 const int32_t traversalIdx = traversalOrder[i] - origRank; in prepare()
239 destDims[origDim] *= dimensions[i]; in prepare()