Home
last modified time | relevance | path

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

/frameworks/ml/nn/common/operations/
DQuantizedLSTM.h79 const RunTimeOperandInfo* cellGateBias_; variable
DQuantizedLSTMTest.cpp128 ASSERT_EQ(setInputTensor(&execution, QuantizedLSTMCell::kCellGateBiasTensor, cellGateBias_), in invoke()
180 cellGateBias_ = cellGateBias; in setWeightsAndBiases()
213 std::vector<int32_t> cellGateBias_; member in android::nn::wrapper::QuantizedLSTMOpModel
DQuantizedLSTM.cpp238 cellGateBias_ = GetInput(operation, operands, kCellGateBiasTensor); in QuantizedLSTMCell()
370 memcpy(bias + 1 * outputSize, GetBuffer<int32_t>(cellGateBias_), sizeof(int32_t) * outputSize); in concatenateBiases()