Home
last modified time | relevance | path

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

/frameworks/ml/nn/common/operations/
DQuantizedLSTM.h69 const RunTimeOperandInfo* inputToCellWeights_; variable
DQuantizedLSTMTest.cpp105 inputToCellWeights_), in invoke()
172 inputToCellWeights_ = inputToCellWeights; in setWeightsAndBiases()
205 std::vector<uint8_t> inputToCellWeights_; member in android::nn::wrapper::QuantizedLSTMOpModel
DQuantizedLSTM.cpp228 inputToCellWeights_ = GetInput(operation, operands, kInputToCellWeightsTensor); in QuantizedLSTMCell()
357 assignWeightsSubmatrix(inputToCellWeights_, 1 * outputSize, outputSize, weightsDims, weights); in concatenateWeights()