Searched refs:weightsDims (Results 1 – 3 of 3) sorted by relevance
| /packages/modules/NeuralNetworks/common/cpu_operations/ |
| D | QuantizedLSTM.cpp | 211 const int32_t offset_column, const std::vector<uint32_t>& weightsDims, in assignWeightsSubmatrix() argument
218 weights[(row + offset_row) * weightsDims[1] + column + offset_column] = submatrixValues[i]; in assignWeightsSubmatrix()
353 void QuantizedLSTMCell::concatenateWeights(const std::vector<uint32_t>& weightsDims, in concatenateWeights() argument
357 assignWeightsSubmatrix(inputToInputWeights_, 0 * outputSize, outputSize, weightsDims, weights); in concatenateWeights()
358 assignWeightsSubmatrix(inputToCellWeights_, 1 * outputSize, outputSize, weightsDims, weights); in concatenateWeights()
359 assignWeightsSubmatrix(inputToForgetWeights_, 2 * outputSize, outputSize, weightsDims, weights); in concatenateWeights()
360 assignWeightsSubmatrix(inputToOutputWeights_, 3 * outputSize, outputSize, weightsDims, weights); in concatenateWeights()
361 assignWeightsSubmatrix(recurrentToInputWeights_, 0 * outputSize, 0, weightsDims, weights); in concatenateWeights()
362 assignWeightsSubmatrix(recurrentToCellWeights_, 1 * outputSize, 0, weightsDims, weights); in concatenateWeights()
363 assignWeightsSubmatrix(recurrentToForgetWeights_, 2 * outputSize, 0, weightsDims, weights); in concatenateWeights()
[all …]
|
| /packages/modules/NeuralNetworks/common/types/operations/include/ |
| D | QuantizedLSTM.h | 88 void concatenateWeights(const std::vector<uint32_t>& weightsDims, uint8_t* weights);
|
| /packages/modules/NeuralNetworks/runtime/test/ |
| D | TestValidateOperations.cpp | 3970 uint32_t weightsDims[2] = {inputSize, numUnits}; in bidirectionlSequenceRNNTest() local
3983 .dimensions = weightsDims, in bidirectionlSequenceRNNTest()
4046 uint32_t weightsDims[2] = {inputSize, numUnits}; in unidirectionlSequenceRNNTest() local
4059 .dimensions = weightsDims, in unidirectionlSequenceRNNTest()
|