Lines Matching refs:indices
47 uint64_t getFlattenedIndex(const std::vector<int32_t>& indices, const std::vector<uint32_t>& shape, in getFlattenedIndex() argument
53 index += uint64_t(indices[i] * subElems); in getFlattenedIndex()
90 void populate(const T* srcData, std::vector<int32_t>* indices, uint32_t level, uint32_t prevIdx, in populate() argument
95 if (level == (*indices).size()) { // level == size of traversal order in populate()
101 origIdx[origDim] = (*indices)[i]; in populate()
104 for (; i < (*indices).size(); i++) { in populate()
107 origIdx[origDim] = origIdx[origDim] * blockSize[blockIdx] + (*indices)[i]; in populate()
117 (*indices)[level] = i; in populate()
118 populate(srcData, indices, level + 1, prevIdx * shapeOfLevel + i, destData, destDims, in populate()
125 (*indices)[level] = arrayIndices[i]; in populate()
126 populate(srcData, indices, level + 1, i, destData, destDims, dimFormat, traversalOrder, in populate()
212 std::vector<int32_t> indices( in densify() local
214 populate(srcData, &indices, 0u, 0u, destData, destShape.dimensions, dimFormat, traversalOrder, in densify()