Home
last modified time | relevance | path

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

/frameworks/ml/nn/common/operations/
DQuantizedLSTMTest.cpp80 &prevCellState_); in QuantizedLSTMOpModel()
134 setInputTensor(&execution, QuantizedLSTMCell::kPrevCellStateTensor, prevCellState_), in invoke()
149 prevCellState_ = cellStateOut_; in invoke()
215 std::vector<int16_t> prevCellState_; member in android::nn::wrapper::QuantizedLSTMOpModel
DQuantizedLSTM.h82 const RunTimeOperandInfo* prevCellState_; variable
DQuantizedLSTM.cpp241 prevCellState_ = GetInput(operation, operands, kPrevCellStateTensor); in QuantizedLSTMCell()
418 convertShapeToDims(biasShape), GetBuffer<const int16_t>(prevCellState_), in eval()
419 convertShapeToDims(prevCellState_->shape()), in eval()