Home
last modified time | relevance | path

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

/aosp12/packages/modules/NeuralNetworks/common/operations/
H A DQuantizedLSTMTest.cpp129 ASSERT_EQ(setInputTensor(&execution, QuantizedLSTMCell::kCellGateBiasTensor, cellGateBias_), in invoke()
181 cellGateBias_ = cellGateBias; in setWeightsAndBiases()
214 std::vector<int32_t> cellGateBias_; member in android::nn::wrapper::QuantizedLSTMOpModel
H A DQuantizedLSTM.h79 const RunTimeOperandInfo* cellGateBias_; variable
H A DQuantizedLSTM.cpp235 cellGateBias_ = GetInput(operation, operands, kCellGateBiasTensor); in QuantizedLSTMCell()
367 memcpy(bias + 1 * outputSize, GetBuffer<int32_t>(cellGateBias_), sizeof(int32_t) * outputSize); in concatenateBiases()