Home
last modified time | relevance | path

Searched refs:submatrixDims (Results 1 – 1 of 1) sorted by relevance

/aosp12/packages/modules/NeuralNetworks/common/operations/
H A DQuantizedLSTM.cpp210 const std::vector<uint32_t> submatrixDims = submatrix->shape().dimensions; in assignWeightsSubmatrix() local
211 for (uint32_t i = 0; i < submatrixDims[0] * submatrixDims[1]; ++i) { in assignWeightsSubmatrix()
212 const uint32_t row = i / submatrixDims[1]; in assignWeightsSubmatrix()
213 const uint32_t column = i % submatrixDims[1]; in assignWeightsSubmatrix()