Home
last modified time | relevance | path

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

/frameworks/ml/nn/common/operations/
DQuantizedLSTM.cpp210 const int32_t offset_column, const std::vector<uint32_t>& weightsDims, in assignWeightsSubmatrix() argument
217 weights[(row + offset_row) * weightsDims[1] + column + offset_column] = submatrixValues[i]; in assignWeightsSubmatrix()
352 void QuantizedLSTMCell::concatenateWeights(const std::vector<uint32_t>& weightsDims, in concatenateWeights() argument
356 assignWeightsSubmatrix(inputToInputWeights_, 0 * outputSize, outputSize, weightsDims, weights); in concatenateWeights()
357 assignWeightsSubmatrix(inputToCellWeights_, 1 * outputSize, outputSize, weightsDims, weights); in concatenateWeights()
358 assignWeightsSubmatrix(inputToForgetWeights_, 2 * outputSize, outputSize, weightsDims, weights); in concatenateWeights()
359 assignWeightsSubmatrix(inputToOutputWeights_, 3 * outputSize, outputSize, weightsDims, weights); in concatenateWeights()
360 assignWeightsSubmatrix(recurrentToInputWeights_, 0 * outputSize, 0, weightsDims, weights); in concatenateWeights()
361 assignWeightsSubmatrix(recurrentToCellWeights_, 1 * outputSize, 0, weightsDims, weights); in concatenateWeights()
362 assignWeightsSubmatrix(recurrentToForgetWeights_, 2 * outputSize, 0, weightsDims, weights); in concatenateWeights()
[all …]
DQuantizedLSTM.h88 void concatenateWeights(const std::vector<uint32_t>& weightsDims, uint8_t* weights);
/frameworks/ml/nn/runtime/test/
DTestValidateOperations.cpp3932 uint32_t weightsDims[2] = {inputSize, numUnits}; in bidirectionlSequenceRNNTest() local
3945 .dimensions = weightsDims, in bidirectionlSequenceRNNTest()
4008 uint32_t weightsDims[2] = {inputSize, numUnits}; in unidirectionlSequenceRNNTest() local
4021 .dimensions = weightsDims, in unidirectionlSequenceRNNTest()