Searched refs:cellStateOut_ (Results 1 – 3 of 3) sorted by relevance
83 cellStateOut_.resize(numBatches * outputSize, 0); in QuantizedLSTMOpModel()141 setOutputTensor(&execution, QuantizedLSTMCell::kCellStateOutTensor, &cellStateOut_), in invoke()150 prevCellState_ = cellStateOut_; in invoke()219 std::vector<int16_t> cellStateOut_; member in android::nn::wrapper::QuantizedLSTMOpModel
85 RunTimeOperandInfo* cellStateOut_; variable
241 cellStateOut_ = GetOutput(operation, operands, kCellStateOutTensor); in QuantizedLSTMCell()418 GetBuffer<int16_t>(cellStateOut_), convertShapeToDims(cellStateOut_->shape()), in eval()