Home
last modified time | relevance | path

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

/packages/modules/NeuralNetworks/common/cpu_operations/
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
DQuantizedLSTM.cpp239 cellGateBias_ = GetInput(operation, operands, kCellGateBiasTensor); in QuantizedLSTMCell()
371 memcpy(bias + 1 * outputSize, GetBuffer<int32_t>(cellGateBias_), sizeof(int32_t) * outputSize); in concatenateBiases()
/packages/modules/NeuralNetworks/common/types/operations/include/
DQuantizedLSTM.h79 const RunTimeOperandInfo* cellGateBias_; variable