Searched refs:origRank (Results 1 – 1 of 1) sorted by relevance
| /packages/modules/NeuralNetworks/common/cpu_operations/ |
| D | Densify.cpp | 48 const int origRank) { in getFlattenedIndex() argument
52 for (int i = origRank - 1; i >= 0; i--) { in getFlattenedIndex()
94 const std::vector<std::vector<int32_t>>& dimMetadata, const int origRank) { in populate() argument
96 std::vector<int> origIdx(origRank); in populate()
105 const int blockIdx = traversalOrder[i] - origRank; in populate()
110 destData[getFlattenedIndex(origIdx, destDims, origRank)] = srcData[prevIdx]; in populate()
119 dimFormat, traversalOrder, blockSize, blockMap, dimMetadata, origRank); in populate()
127 blockSize, blockMap, dimMetadata, origRank); in populate()
187 const int origRank = destShape.dimensions.size(); in densify() local
191 const int32_t origDim = traversalOrder[origRank + i]; in densify()
[all …]
|