Home
last modified time | relevance | path

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

/frameworks/ml/nn/common/operations/
DQuantizedLSTM.h73 const RunTimeOperandInfo* recurrentToForgetWeights_; variable
DQuantizedLSTMTest.cpp114 recurrentToForgetWeights_), in invoke()
175 recurrentToForgetWeights_ = recurrentToForgetWeights; in setWeightsAndBiases()
208 std::vector<uint8_t> recurrentToForgetWeights_; member in android::nn::wrapper::QuantizedLSTMOpModel
DQuantizedLSTM.cpp232 recurrentToForgetWeights_ = GetInput(operation, operands, kRecurrentToForgetWeightsTensor); in QuantizedLSTMCell()
362 assignWeightsSubmatrix(recurrentToForgetWeights_, 2 * outputSize, 0, weightsDims, weights); in concatenateWeights()